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()