Update app.py

app.py

@@ -5,12 +5,14 @@ import indicnlp
 from indicnlp.transliterate.unicode_transliterate import UnicodeIndicTransliterator
 
 model_checkpoint = './hindi_sanskrit'
-translator = pipeline('translation', model=model_checkpoint, decoder_start_token='<2ka>')
+translator = pipeline('translation', model=model_checkpoint, decoder_start_token='<2sa>')
+
+st.title('Hindi to Sanskrit Translation')
 
 query = st.text_input("Enter the Input Sentence", "")
 
 input_sentence = query + ' </s>' + ' <2hi>'
 output_sentence = translator(input_sentence,max_length=128)
-#output_sentence = UnicodeIndicTransliterator.transliterate(output_sentence[0]['translation_text'],'hi','kn')
+output_sentence = output_sentence[0]["translation_text"]
 
-st.write(output_sentence)
+st.write(output_sentence)