Update app.py

app.py

@@ -1,5 +1,4 @@
 # prompt: write a streamlit app that converts english text into french text when translate button is pressed. The title of page should be "Translate to French"."
-
 import streamlit as st
 from transformers import pipeline
 
@@ -8,6 +7,11 @@ st.title("Translate to French")
 input_text = st.text_input("Enter text to translate")
 
 if st.button("Translate"):
-    pipe = pipeline(model="facebook/mbart-large-cc25")
+    # 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)