Update app.py

app.py

@@ -5,7 +5,6 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline
 model_name = "Helsinki-NLP/opus-mt-en-fr"  # Example model for English to French
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-translator = pipeline("translation", model=model, tokenizer=tokenizer)
 
 # Define the translation function
 def translate_text(text, target_language):
@@ -23,17 +22,22 @@ languages = {
     "ru": "Russian"
 }
 
-# Set up the Gradio interface
-iface = gr.Interface(
-    fn=translate_text,
-    inputs=[
-        gr.Textbox(label="Enter text to translate"),
-        gr.Dropdown(list(languages.keys()), label="Target Language", type="value")
-    ],
-    outputs=gr.Textbox(label="Translation"),
-    live=True,
-    title="Text Translator",
-    description="Translate text into multiple languages using Hugging Face models."
-)
+# Set up the Gradio interface with a submit button
+with gr.Blocks() as iface:
+    gr.Markdown("# Text Translator")
+    gr.Markdown("Translate text into multiple languages using Hugging Face models.")
+
+    # Input components
+    text_input = gr.Textbox(label="Enter text to translate")
+    language_dropdown = gr.Dropdown(list(languages.keys()), label="Target Language", type="value")
+
+    # Output component
+    translation_output = gr.Textbox(label="Translation")
+
+    # Button to submit
+    submit_button = gr.Button("Translate")
+
+    # When button is clicked, trigger the translation
+    submit_button.click(fn=translate_text, inputs=[text_input, language_dropdown], outputs=translation_output)
 
 iface.launch()