Spaces:
Runtime error
Runtime error
File size: 689 Bytes
9863176 b2774fc dbea2e3 9863176 b2774fc 670b28d 9863176 cf11ee6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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='<2sa>')
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('Sanskrit Sentence:',output_sentence) |