amnashoukat commited on
Commit
e57b72c
·
verified ·
1 Parent(s): 50d7def

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +45 -0
app.py ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Install dependencies (Run this in Google Colab),!pip install transformers gradio --quiet
2
+
3
+ # Import libraries
4
+ from transformers import MarianMTModel, MarianTokenizer
5
+ import gradio as gr
6
+
7
+ # Define model names for English↔Urdu translation
8
+ en_to_ur_model_name = "Helsinki-NLP/opus-mt-en-ur"
9
+ ur_to_en_model_name = "Helsinki-NLP/opus-mt-ur-en"
10
+
11
+ # Load models and tokenizers
12
+ en_to_ur_tokenizer = MarianTokenizer.from_pretrained(en_to_ur_model_name)
13
+ en_to_ur_model = MarianMTModel.from_pretrained(en_to_ur_model_name)
14
+
15
+ ur_to_en_tokenizer = MarianTokenizer.from_pretrained(ur_to_en_model_name)
16
+ ur_to_en_model = MarianMTModel.from_pretrained(ur_to_en_model_name)
17
+
18
+ # Translation functions
19
+ def translate_en_to_ur(text):
20
+ inputs = en_to_ur_tokenizer(text, return_tensors="pt", padding=True, truncation=True)
21
+ translated = en_to_ur_model.generate(**inputs)
22
+ return en_to_ur_tokenizer.decode(translated[0], skip_special_tokens=True)
23
+
24
+ def translate_ur_to_en(text):
25
+ inputs = ur_to_en_tokenizer(text, return_tensors="pt", padding=True, truncation=True)
26
+ translated = ur_to_en_model.generate(**inputs)
27
+ return ur_to_en_tokenizer.decode(translated[0], skip_special_tokens=True)
28
+
29
+ # Gradio interface
30
+ with gr.Blocks() as demo:
31
+ gr.Markdown("## 🈯 English ↔ Urdu Translator")
32
+ with gr.Tab("English ➜ Urdu"):
33
+ en_input = gr.Textbox(label="Enter English Text")
34
+ en_output = gr.Textbox(label="Translated Urdu Text")
35
+ en_translate_btn = gr.Button("Translate to Urdu")
36
+ en_translate_btn.click(translate_en_to_ur, en_input, en_output)
37
+
38
+ with gr.Tab("Urdu ➜ English"):
39
+ ur_input = gr.Textbox(label="اردو متن داخل کریں")
40
+ ur_output = gr.Textbox(label="Translated English Text")
41
+ ur_translate_btn = gr.Button("Translate to English")
42
+ ur_translate_btn.click(translate_ur_to_en, ur_input, ur_output)
43
+
44
+ # Launch the app (for local testing or in Colab)
45
+ demo.launch()