import gradio as gr
import subprocess
subprocess.check_call(["pip", "install", "transformers"])
subprocess.check_call(["pip", "install", "torch"])

from transformers import pipeline

pipe = pipeline("text2text-generation", model="balaramas/mbart-sahitrans_new_data")


def sanmt(txt):
    output=pipe(txt, max_length=20, min_length=5, do_sample=False)[0]['generated_text']  
    return output

iface = gr.Interface(
    fn=sanmt, 
    inputs=gr.Textbox(label="Enter text in Sanskrit", placeholder="Type here..."), 
    outputs=gr.Textbox(label="Translated Hindi Text"),
    title="Sanskrit to Hindi Translator"
)
iface.launch()