Update app.py

app.py

@@ -26,6 +26,7 @@ model_trans = AutoModelForCausalLM.from_pretrained(
 
 # --- Sửa đổi hàm translate_text ---
 # Thêm tham số model_name (mặc định là "quickmt-en-vi")
+@spaces.GPU(duration=60)
 def translate_text(text, lang=None, progress=gr.Progress(track_tqdm=True)):
   # Set default values if None (happens during example caching)
     if lang is None:
@@ -201,7 +202,13 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
                 label="",
                 placeholder="",
                 lines=1
-            )           
+            )   
+
+            text_translate = gr.Textbox(
+                label="",
+                placeholder="",
+                lines=1
+            )                 
             
             with gr.Row():
                 clear_button = gr.Button("Clear Chat")
@@ -225,6 +232,12 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         outputs=[prompt_box, chatbot_display, internal_history, response, prompt]
     )
 
+    text_translate.submit(
+        fn=translate_text,
+        inputs=[text_translate,lang],
+        outputs=[prompt]
+    )
+
     # Khi người dùng nhấn nút "Send" (ẩn)
     submit_button.click(
         fn=chat_with_model,