Update app.py

app.py

@@ -2,15 +2,11 @@
 import streamlit as st
 from transformers import pipeline
 
-st.title("Translate to French")
-input_text = st.text_input("Enter text to translate")
+def main():
+    st.title("Translate to French")
+    sentence = st.text_input("Enter your sentence:")
+    if st.button("Translate"):
+        pipe = pipeline(model="facebook/m2m100_418M")
+        translation = pipe(sentence, target_lang="fr")[0]["translation_text"]
+        st.write(translation)
 
-if st.button("Translate"):
-    # Specify the translation task explicitly
-    pipe = pipeline(task="translation", model="facebook/mbart-large-cc25", tokenizer="facebook/mbart-large-cc25")
-    
-    # Perform translation
-    translation = pipe(input_text, target_lang="fr")[0]["translation_text"]
-    
-    # Display the translated text
-    st.write(translation)