Update app.py

app.py

@@ -14,29 +14,29 @@ text = st.text_area('Enter the text:')
 
 if text:
     model_inputs = tokenizer_translation(text, return_tensors="pt")
-    lg = st.number_input("Select Language: 1.Hindi, 2.Telugu, 3.Gujarati, 4.Bengali")
-    if lg==1:
+    lg = st.text_input("Select Language: hi.Hindi, te.Telugu, gu.Gujarati, bn.Bengali")
+    if lg=='hi':
         generated_tokens = model_translation.generate(
         **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:
+        st.write(translation)
+    elif lg=='te':
         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:
+        st.write(translation)
+    elif lg=='gu':
         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:
+        st.write(translation)
+    elif lg=='bn':
         generated_tokens = model_translation.generate(
         **model_inputs,
         forced_bos_token_id=tokenizer_translation.lang_code_to_id["bn_IN"]