Update app.py

app.py

@@ -4,6 +4,12 @@ import torch
 from PIL import Image
 from transformers import AutoModel, AutoTokenizer
 import spaces
+from googletrans import Translator
+
+translator = Translator()
+
+def translate_to_english(text):
+    return translator.translate(text, src='zh-cn', dest='en').text
 
 device="cuda"
 
@@ -16,6 +22,23 @@ model.eval()
 # Define a function to generate a response
 @spaces.GPU
 def generate_response(image, question):
+    msgs = [{'role': 'user', 'content': question}]
+    res = model.chat(
+        image=image,
+        msgs=msgs,
+        tokenizer=tokenizer,
+        sampling=True,
+        temperature=0.7,
+        stream=True
+    )
+    generated_text = "".join(res)
+    
+    # Translate to English
+    translated_text = translate_to_english(generated_text)
+    
+    return translated_text
+
+def generate_response_old(image, question):
     msgs = [{'role': 'user', 'content': question}]
     res = model.chat(
         image=image,
@@ -33,11 +56,8 @@ def generate_response(image, question):
 # Create the footer with links
 footer = """
 <div style="text-align: center; margin-top: 20px;">
-    <a href="https://www.linkedin.com/in/pejman-ebrahimi-4a60151a7/" target="_blank">LinkedIn</a> |
-    <a href="https://github.com/arad1367/Visual_QA_MiniCPM-Llama3-V-2_5_GradioApp" target="_blank">GitHub</a> |
-    <a href="https://arad1367.pythonanywhere.com/" target="_blank">Live demo of my PhD defense</a>
     <br>
-    Made with 💖 by Pejman Ebrahimi
+    Made with 💖 by CodeMonkeyXL
 </div>
 """
 
@@ -52,4 +72,4 @@ with gr.Blocks(theme='abidlabs/dracula_revamped') as demo:
     gr.HTML(footer)
 
 # Launch the app
-demo.launch(debug=True)
+demo.launch(debug=True,show_api=True)