Update app.py

app.py

@@ -1,18 +1,23 @@
 import gradio as gr
-from indic_transliteration import sanscript
-from indic_transliteration.sanscript import transliterate
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-def translit_text(text):
-    # Roman Hindi से Hindi (Devanagari) में बदलना
-    hindi_text = transliterate(text, sanscript.ITRANS, sanscript.DEVANAGARI)
+# Hinglish to Hindi Transliteration model (ai4bharat)
+model_name = "ai4bharat/indictrans2-en-indic-1B"  
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name, trust_remote_code=True)
+
+def transliterate_text(text):
+    inputs = tokenizer(text, return_tensors="pt")
+    outputs = model.generate(**inputs, max_length=128)
+    hindi_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return hindi_text
 
 demo = gr.Interface(
-    fn=translit_text,
+    fn=transliterate_text,
     inputs="text",
     outputs="text",
     title="🔤 Roman Hindi → Hindi Transliteration Demo",
-    description="Type in Roman Hindi (like: mujhe pyar hai) and see it in Devanagari Hindi."
+    description="Type in Roman Hindi (like: mujhe pyar hai) and see it in proper Hindi."
 )
 
 if __name__ == "__main__":