Update app.py

app.py

@@ -22,10 +22,9 @@ if submit_button:
 
     # Perform translation
     output_ids = model.generate(input_ids)
-    translated = tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
     # Decode the translated text
     translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
     # Display the translated text
-    st.write("Translated Text:", translated_text[0])
+    st.write("Translated Text:", translated_text)