app.py

import gradio as gr
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("salesken/translation-hi-en")
model = AutoModelForSeq2SeqLM.from_pretrained("salesken/translation-hi-en")

def greet(name):
    hin_snippet = name
    inputs = tokenizer.encode(hin_snippet, return_tensors="pt",padding=True,max_length=512,truncation=True)
    outputs = model.generate(inputs, max_length=128, num_beams=None, early_stopping=True)
    translated = tokenizer.decode(outputs[0]).replace('<pad>',"").strip().lower()
    return translated

iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()