Update app.py

app.py

@@ -16,6 +16,41 @@ session_token = os.environ.get('SessionToken')
 
 
 
+count = 0
+
+def translate_language(text_prompts):
+    global count
+    try:
+        count += 1
+        tips_text = None
+        language_code = language_recognition_model.recognize(text_prompts)
+        if language_code != 'zh':
+            text_prompts = language_translation_model.translate(text_prompts, language_code, 'zh')
+    except Exception as e:
+        error_text = str(e)
+        return {status_text:error_text, language_tips_text:gr.update(visible=False)}
+    if language_code in tips:
+        tips_text = tips[language_code]
+    else:
+        tips_text = tips['en']
+    if language_code == 'zh':
+        return {language_tips_text:gr.update(visible=False), translated_language:text_prompts, trigger_component: gr.update(value=count, visible=False)}
+    else:
+        return {language_tips_text:gr.update(visible=True, value=tips_text), translated_language:text_prompts, trigger_component:  gr.update(value=count, visible=False)}
+
+        
+def inference(text_prompts, style_indx):
+  try:
+    model.token = model._apply_token(model.ak, model.sk)
+    style = style_list[style_indx]
+    results = model.generate_image(
+        text_prompts=text_prompts, style=style, visualization=False, topk=4)
+  except Exception as e:
+    error_text = str(e)
+    return {status_text:error_text, gallery:None}
+  return {status_text:'Success', gallery:results[:4]}
+
+
 
 title="بسم الله الرحمن الرحيم"
 description="ERNIE-ViLG model, which supports text-to-image task."
@@ -134,7 +169,6 @@ with block:
         """
     )
 
-description="ERNIE-ViLG model, which supports text-to-image task."