Update app.py

app.py

@@ -32,9 +32,9 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
     else:
         prompt = f"translate {sselected_language} to {tselected_language}: {input_text}"
 
-    input_ids = tokenizer.encode(prompt, return_tensors="pt")
-    output_ids = model.generate(input_ids)
-    translated_text = tokenizer.decode(**output_ids, skip_special_tokens=True)
+    input_ids = tokenizer.encode([prompt], return_tensors="pt")
+    output_ids = model.generate(**input_ids)
+    translated_text = tokenizer.batch_decode(**output_ids, skip_special_tokens=True)[0]
 
     return translated_text