IndicBART / app.py
Rohan's picture
Upload app.py
9863176
import streamlit as st
import transformers
from transformers import pipeline
import indicnlp
from indicnlp.transliterate.unicode_transliterate import UnicodeIndicTransliterator
model_checkpoint = './hindi_sanskrit'
translator = pipeline('translation', model=model_checkpoint, decoder_start_token='<2ka>')
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')
st.write(output_sentence)