Merge branch 'main' of https://huggingface.co/spaces/yizhangliu/Text-to-Image

app.py

@@ -7,27 +7,31 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from loguru import logger
 
+from utils import get_tmt_client, getTextTrans_tmt
 import subprocess
 
 result = subprocess.run(['pip', 'list'], check=True)
 print(f'pip list = {result}')
 
-language_translation_model = hub.Module(directory=f'./baidu_translate')
+tmt_client = get_tmt_client()
+
+# language_translation_model = hub.Module(directory=f'./baidu_translate')
 def getTextTrans(text, source='zh', target='en'):
-    def is_chinese(string):
-        for ch in string:
-            if u'\u4e00' <= ch <= u'\u9fff':
-                return True
-        return False
+    return getTextTrans_tmt(tmt_client, text, source, target)
+    # def is_chinese(string):
+    #     for ch in string:
+    #         if u'\u4e00' <= ch <= u'\u9fff':
+    #             return True
+    #     return False
         
-    if not is_chinese(text) and target == 'en': 
-        return text
+    # if not is_chinese(text) and target == 'en': 
+    #     return text
         
-    try:
-        text_translation = language_translation_model.translate(text, source, target)
-        return text_translation
-    except Exception as e:
-        return text 
+    # try:
+    #     text_translation = language_translation_model.translate(text, source, target)
+    #     return text_translation
+    # except Exception as e:
+    #     return text 
         
 space_ids = {
             "spaces/stabilityai/stable-diffusion": "SD 2.1",
@@ -392,5 +396,4 @@ with gr.Blocks(title='Text-to-Image') as demo:
                 draw_btn_0.click(fn=prompt_draw_2, inputs=[prompt_input0], outputs=[prompt_work, prompt_work_zh])
                 draw_btn_1.click(fn=prompt_draw_2, inputs=[prompt_input1], outputs=[prompt_work, prompt_work_zh])                
 
-demo.queue()
-demo.launch()
+demo.queue(concurrency_count=128).launch()