Update app.py

app.py

@@ -1,14 +1,22 @@
-import streamlit as st 
-import transformers 
+import streamlit as st
 from transformers import pipeline
 
-st.title("Text Generation with Hugging Face")
+# Load the translation pipeline
+translator = pipeline("translation_en_to_es", model="Helsinki-NLP/opus-mt-en-es")
 
-model_name = "bigscience/mt0-xxl-mt"
-generator = pipeline("text-generation", model=model_name, device=0)
+# Set the Streamlit app title
+st.title("Machine Translation with Hugging Face")
 
-prompt = st.text_input("Enter text prompt:", "Once upon a time")
+# Define the input text area
+input_text = st.text_area("Enter text in English to translate to Spanish:", height=200)
+
+# Define the translate button
+translate_button = st.button("Translate")
+
+# If the translate button is clicked
+if translate_button:
+    # Translate the input text
+    output_text = translator(input_text, max_length=1000)[0]['translation_text']
+    # Display the output text
+    st.text_area("Translated text in Spanish:", value=output_text, height=200)
 
-if st.button("Generate Text"):
-    output = generator(prompt, max_length=100, do_sample=True)
-    st.write(output[0]['generated_text'])