alpha2002 commited on
Commit
ffc61d1
·
verified ·
1 Parent(s): b301e03

create english_to_darija_gradio.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+ from transformers import pipeline
4
+
5
+ # Initialize the translation pipeline
6
+ pipe = pipeline("translation", model="alpha2002/eng_alpha_darija", tokenizer="alpha2002/eng_alpha_darija")
7
+
8
+ def translate_transformers(from_text):
9
+ translation = pipe(from_text, src_lang="en_XX", tgt_lang="ar_AR")
10
+ return translation[0]['translation_text']
11
+
12
+ def translate_gradio(input_text):
13
+ return translate_transformers(input_text)
14
+
15
+ interface = gr.Interface(fn=translate_gradio,
16
+ title="English to Darija Translator",
17
+ description="Enter English text to translate it to Darija in Arabic script.",
18
+ inputs="text",
19
+ outputs='text')
20
+
21
+ interface.launch()