Create app.py

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+model_translate = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M")
+tokenizer_translate = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
+inlang='en'
+outlang='hi'
+tokenizer_translate.src_lang = inlang
+
+def translate(text):    
+    encoded_en = tokenizer_translate(text, return_tensors="pt")
+    generated_tokens = model_translate.generate(**encoded_en, forced_bos_token_id=tokenizer_translate.get_lang_id(outlang))
+    return tokenizer_translate.batch_decode(generated_tokens, skip_special_tokens=True)[0]
+    
+interface = gr.Interface(
+    fn = translate,
+    inputs = 'text',
+    outputs = 'text'
+)
+
+interface.launch()