Update app.py

app.py

@@ -128,16 +128,14 @@ 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")
-        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 encoded_input
+        translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
+    return translated_text
 
 def print_s(source_lang, target_lang, text0):
     print(source_lang)