Update app.py

app.py

@@ -133,7 +133,7 @@ def translation_text(source_lang, target_lang, user_input):
     with torch.no_grad():
         encoded_input = tokenizer(user_input, return_tensors="pt")
         print(encoded_input)
-        generated_tokens = translation_model.generate(encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
+        generated_tokens = translation_model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
         print(generated_tokens)
         translated_text = tokenizer.batch_decode(
         generated_tokens, skip_special_tokens=True)[0]