Update app.py

app.py

@@ -7,12 +7,15 @@ from indicnlp.transliterate.unicode_transliterate import UnicodeIndicTranslitera
 model_checkpoint = './hindi_sanskrit'
 translator = pipeline('translation', model=model_checkpoint, decoder_start_token='<2sa>')
 
-st.title('Hindi to Sanskrit Translation')
+st.title('NLTM')
+
+st.header('Hindi-Sanskrit Translation Model')
 
 query = st.text_input("Enter the Input Sentence", "")
 
 input_sentence = query + ' </s>' + ' <2hi>'
 output_sentence = translator(input_sentence,max_length=128)
 output_sentence = output_sentence[0]["translation_text"]
+output_sentence = output_sentence.replace('<2sa'>,'')
 
 st.write(output_sentence)