Update app.py

app.py

@@ -65,13 +65,38 @@ agent = initialize_agent(
     )
 )
 
+from langchain.chains import LLMChain
+from langchain.prompts import PromptTemplate
+def enToChinese(english):
+    #ch = llm_fy("Please translate the following sentence from English to Chinese："+english)
+    #return ch
+    pp = "Please translate the following sentence from English to Chinese：{english}"
+    prompt = PromptTemplate(
+           input_variables=["english"],
+           template=pp
+    )
+    llchain=LLMChain(llm=llm,prompt=prompt)
+    return llchain.run(english)
+    
+
+def chToEnglish(chinese):
+    #en = llm_fy("Please translate the following sentence from Chinese to English："+chinese)
+    #return en
+    pp = "Please translate the following sentence from Chinese to English：{chinese}"
+    prompt = PromptTemplate(
+           input_variables=["chinese"],
+           template=pp
+    )
+    llchain=LLMChain(llm=llm,prompt=prompt)
+    return llchain.run(chinese)
+
 import gradio as gr
 def segment(image,text):
     #pass  # Implement your image segmentation model here...
     print(image)
     image_url = image
-    
-    return agent(f"{text}:\n{image_url}").get('output').replace('The response to your last comment is','')
+    text = chToEnglish(text)
+    return enToChinese(agent(f"{text}:\n{image_url}").get('output'))
 
 demo = gr.Interface(
     fn=segment,