Update app.py

app.py

@@ -128,10 +128,13 @@ lang_id = {
 def translation_text(source_lang, target_lang, user_input):
     src_lang = lang_id[source_lang]
     trg_lang = lang_id[target_lang]
+    print(trg_lang)
     tokenizer.src_lang = src_lang
     with torch.no_grad():
-        encoded_input = tokenizer(user_input, return_tensors="pt").to(device)
+        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))
+        print(generated_tokens)
         translated_text = tokenizer.batch_decode(
         generated_tokens, skip_special_tokens=True)[0]
     return translated_text