Upload 4 files

README.md

@@ -9,6 +9,7 @@ app_file: app.py
 pinned: false
 license: apache-2.0
 header: mini
+hf_oauth: true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -70,7 +70,9 @@ def main():
                 with gr.Group():
                     output_text_pony = gr.TextArea(label="Output tags (Pony e621 style)", interactive=False, show_copy_button=True)
                     copy_btn_pony = gr.Button(value="Copy to clipboard", size="sm", interactive=False)
-            description_ui2()
+        description_ui2()
+        gr.LoginButton()
+        gr.DuplicateButton(value="Duplicate Space for private use")
 
         random_character.click(select_random_character, [input_copyright, input_character], [input_copyright, input_character], queue=False)
 

requirements.txt

@@ -5,8 +5,6 @@ transformers
 optimum[onnxruntime]
 spaces
 dartrs
-httpx==0.13.3
-httpcore
-googletrans==4.0.0rc1
+translatepy
 sentencepiece
 timm

tagger.py

@@ -1,11 +1,8 @@
+import spaces
 from PIL import Image
 import torch
 import gradio as gr
-import spaces
-from transformers import (
-    AutoImageProcessor,
-    AutoModelForImageClassification,
-)
+from transformers import AutoImageProcessor, AutoModelForImageClassification
 from pathlib import Path
 
 
@@ -190,18 +187,16 @@ def convert_danbooru_to_e621_prompt(input_prompt: str = "", prompt_type: str = "
     return output_prompt
 
 
-def translate_prompt(prompt: str = ""):
-    def translate_to_english(prompt):
-        import httpcore
-        setattr(httpcore, 'SyncHTTPTransport', 'AsyncHTTPProxy')
-        from googletrans import Translator
-        translator = Translator()
+from translatepy import Translator
+translator = Translator()
+def translate_prompt_old(prompt: str = ""):
+    def translate_to_english(input: str):
         try:
-            translated_prompt = translator.translate(prompt, src='auto', dest='en').text
-            return translated_prompt
+            output = str(translator.translate(input, 'English'))
         except Exception as e:
+            output = input
             print(e)
-            return prompt
+        return output
 
     def is_japanese(s):
         import unicodedata
@@ -223,18 +218,23 @@ def translate_prompt(prompt: str = ""):
     return ", ".join(outputs)
 
 
+def translate_prompt(input: str):
+    try:
+        output = str(translator.translate(input, 'English'))
+    except Exception as e:
+        output = input
+        print(e)
+    return output
+
+
 def translate_prompt_to_ja(prompt: str = ""):
-    def translate_to_japanese(prompt):
-        import httpcore
-        setattr(httpcore, 'SyncHTTPTransport', 'AsyncHTTPProxy')
-        from googletrans import Translator
-        translator = Translator()
+    def translate_to_japanese(input: str):
         try:
-            translated_prompt = translator.translate(prompt, src='en', dest='ja').text
-            return translated_prompt
+            output = str(translator.translate(input, 'Japanese'))
         except Exception as e:
+            output = input
             print(e)
-            return prompt
+        return output
 
     def is_japanese(s):
         import unicodedata