Update app.py

app.py

@@ -28,13 +28,13 @@ if submit_button:
     # Encode the input text
     # encoded = tokenizer(text_input, return_tensors="pt")
     
-    input_ids = tokenizer.encode(f'Translate {sselected_language} to {tselected_language}: {input_text}', return_tensors='pt')
+    input_ids = tokenizer.encode(f'translate {sselected_language} to {tselected_language}: {input_text}', return_tensors='pt')
 
     # Perform translation
     output_ids = model.generate(input_ids)
 
     # Decode the translated text
-    translated_text = tokenizer.decode(output_ids, skip_special_tokens=True)
+    translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
     # Display the translated text
     st.write("Translated Text:", translated_text)