File size: 2,520 Bytes
cc92c69 927ee68 cc92c69 927ee68 65d1df8 927ee68 cc92c69 927ee68 cc92c69 e676301 cc92c69 43a4171 4233d1d 43a4171 4233d1d cc92c69 4233d1d cc92c69 4233d1d cc92c69 e676301 43a4171 e676301 cc92c69 e676301 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import streamlit as st
from transformers import pipeline
import string
st.set_page_config(page_title="RLM-Translator", page_icon="🍕", layout="wide", initial_sidebar_state="collapsed")
@st.cache_resource
def load_model():
return pipeline(model="rudrashah/RLM-hinglish-translator")
pipe = load_model()
def check_sentence_end(sentence):
last_char = sentence[-1]
if last_char in string.punctuation:
return sentence
else:
return sentence + "."
def model_rlm(sentence):
text = "mere paas 100 rupaye hain"
text = check_sentence_end(sentence)
template = "Hinglish:\n{hi_en}\n\nEnglish:\n{en}"
english = pipe(template.format(hi_en=text,en=""), max_length=250)
english = english[0]['translation_text']
english = english.replace("<bos>","").replace("<eos>","")
english = english[len(template.format(hi_en=text,en="")):]
return english.strip()
def translate_hinglish_to_english(text):
translated_text = model_rlm(text)
st.session_state['translated_text'] = translated_text
return translated_text
st.title("RLM-Translator")
st.write("A simple Hinglish to English translator. Enter your Hinglish text and click on Translate to get the English translation.")
col1, col2 = st.columns(2)
with col1:
if "current_text" in st.session_state:
hinglish_text = st.text_area("Hinglish", value=st.session_state["current_text"], height=300, key="x", placeholder="something like: aapka naam kya hai?")
else:
hinglish_text = st.text_area("Hinglish", height=300, key="x", placeholder="something like: aapka naam kya hai?")
with col2:
if "translated_text" in st.session_state:
st.text_area("English", value=st.session_state["translated_text"], height=300, key="", placeholder="Translation will appear here.")
else:
st.text_area("English", height=300, key="", placeholder="Translation will appear here.")
if st.button("Translate", use_container_width=True, type="primary"):
if hinglish_text != "":
with st.spinner("Translating..."):
st.session_state["current_text"] = hinglish_text
translated_text = translate_hinglish_to_english(hinglish_text)
st.session_state["translated_text"] = translated_text
st.rerun()
else:
st.error("Please enter some hinglish text to translate.")
st.markdown("*This is in beta Development. Translator might give incorrect results.*")
st.markdown("Made with ❤️ by [Rudra Shah](https://huggingface.co/rudrashah).")
|