Spaces:
Sleeping
Sleeping
| 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() | |