Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-# tokenizer = T5Tokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ro")
+# "Helsinki-NLP/opus-mt-en-ro")
 
 # Create the app layout
 st.title("Text Translation")
@@ -34,7 +34,7 @@ if submit_button:
     output_ids = model.generate(input_ids)
 
     # Decode the translated text
-    translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
+    translated_text = tokenizer.decode(output_ids, skip_special_tokens=True)
 
     # Display the translated text
     st.write("Translated Text:", translated_text)