Update app.py

app.py

@@ -20,23 +20,28 @@ if text:
         **model_inputs,
         forced_bos_token_id=tokenizer_translation.lang_code_to_id["hi_IN"]
         )
+        translation = tokenizer_translation.batch_decode(generated_tokens, skip_special_tokens=True)
+        st.json(translation)
     elif lg==2:
         generated_tokens = model_translation.generate(
         **model_inputs,
         forced_bos_token_id=tokenizer_translation.lang_code_to_id["te_IN"]
         )
+        translation = tokenizer_translation.batch_decode(generated_tokens, skip_special_tokens=True)
+        st.json(translation)
     elif lg==3:
         generated_tokens = model_translation.generate(
         **model_inputs,
         forced_bos_token_id=tokenizer_translation.lang_code_to_id["gu_IN"]
         )
+        translation = tokenizer_translation.batch_decode(generated_tokens, skip_special_tokens=True)
+        st.json(translation)
     elif lg==4:
         generated_tokens = model_translation.generate(
         **model_inputs,
         forced_bos_token_id=tokenizer_translation.lang_code_to_id["bn_IN"]
         )
+        translation = tokenizer_translation.batch_decode(generated_tokens, skip_special_tokens=True)
+        st.json(translation)
     else:
-        return 0
-        
-    translation = tokenizer_translation.batch_decode(generated_tokens, skip_special_tokens=True)
-    st.json(translation)
+        print('invalid choice!')