Rohan commited on
Commit
9863176
1 Parent(s): 3dd2be3

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import transformers
3
+ from transformers import pipeline
4
+ import indicnlp
5
+ from indicnlp.transliterate.unicode_transliterate import UnicodeIndicTransliterator
6
+
7
+ model_checkpoint = './hindi_sanskrit'
8
+ translator = pipeline('translation', model=model_checkpoint, decoder_start_token='<2ka>')
9
+
10
+ query = st.text_input("Enter the Input Sentence", "")
11
+
12
+ input_sentence = query + ' </s>' + ' <2hi>'
13
+ output_sentence = translator(input_sentence,max_length=128)
14
+ #output_sentence = UnicodeIndicTransliterator.transliterate(output_sentence[0]['translation_text'],'hi','kn')
15
+
16
+ st.write(output_sentence)