Update app.py

app.py

@@ -1,16 +1,20 @@
 import gradio as gr
 import numpy as np
 import paddlehub as hub
+from googletrans import Translator
 
 model = hub.Module(name="ernie_vilg")
+translator = Translator()
 
 myInput = gr.Textbox(label="Beschreiwung", value='戴着眼镜的猫')
+
 myOutputs = [
     gr.Gallery(label="Konschtwierk"),
     gr.Textbox(label="Resultat")
     ]
 
 def create(prompt):
+    chinese_translation = translator.translate(prompt, src="lb", dest="cn")
     try:
         artwork = model.generate_image(
             text_prompts=prompt,
@@ -20,7 +24,7 @@ def create(prompt):
     except Exception as e:
         error_text=str(e)
         return None, error_text
-    return artwork, "OK"
+    return artwork, chinese_translation
 
 demo = gr.Interface(
     fn=create,