Bapynshngain's picture
Update app.py
1e0db26 verified
raw
history blame contribute delete
719 Bytes
import gradio as gr
from transformers import MarianMTModel, MarianTokenizer
import os
hf_token = os.environ.get("HF_Bapyn")
model_name = "Bapynshngain/Bapyn-En-Kha"
tokenizer = MarianTokenizer.from_pretrained(model_name, token=hf_token)
model = MarianMTModel.from_pretrained(model_name, token=hf_token)
def translate(text):
inputs = tokenizer(text, return_tensors="pt", padding=True)
translated = model.generate(**inputs)
return tokenizer.decode(translated[0], skip_special_tokens=True)
inp = gr.Textbox(placeholder="Enter English text to translate...")
outp = gr.Textbox(placeholder="Khasi translation")
gr.Interface(fn=translate, inputs=inp, outputs=outp, title="Bapyn Eng-Khasi Translator").launch()