SNA-AI commited on
Commit
62d2049
·
verified ·
1 Parent(s): 12224d1

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import MarianMTModel, MarianTokenizer
3
+
4
+ model_en_ur_name = "Helsinki-NLP/opus-mt-en-ur"
5
+ model_ur_en_name = "Helsinki-NLP/opus-mt-ur-en"
6
+
7
+ tokenizer_en_ur = MarianTokenizer.from_pretrained(model_en_ur_name)
8
+ model_en_ur = MarianMTModel.from_pretrained(model_en_ur_name)
9
+
10
+ tokenizer_ur_en = MarianTokenizer.from_pretrained(model_ur_en_name)
11
+ model_ur_en = MarianMTModel.from_pretrained(model_ur_en_name)
12
+
13
+ def translate_text(text, direction):
14
+ if not text.strip():
15
+ return "⚠️ Please enter some text."
16
+
17
+ if direction == "English → Urdu":
18
+ inputs = tokenizer_en_ur(text, return_tensors="pt", padding=True, truncation=True)
19
+ translated = model_en_ur.generate(**inputs)
20
+ return tokenizer_en_ur.decode(translated[0], skip_special_tokens=True)
21
+
22
+ else:
23
+ inputs = tokenizer_ur_en(text, return_tensors="pt", padding=True, truncation=True)
24
+ translated = model_ur_en.generate(**inputs)
25
+ return tokenizer_ur_en.decode(translated[0], skip_special_tokens=True)
26
+
27
+ with gr.Blocks() as app:
28
+ gr.Markdown("# 🌐 English ↔ Urdu Translator")
29
+ gr.Markdown("Built with Hugging Face + Gradio")
30
+
31
+ text_input = gr.Textbox(lines=4, label="Input Text")
32
+ direction = gr.Radio(["English → Urdu", "Urdu → English"], value="English → Urdu")
33
+ output = gr.Textbox(label="Output")
34
+
35
+ btn = gr.Button("Translate")
36
+ btn.click(translate_text, [text_input, direction], output)
37
+
38
+ app.launch()