vkbajoria commited on
Commit
ed70c4a
1 Parent(s): 30e1e31

creating app.py

Browse files
Files changed (1) hide show
  1. app.py +42 -0
app.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import gradio as gr
3
+
4
+ # Use a pipeline as a high-level helper
5
+ from transformers import pipeline
6
+
7
+ language_codes = {
8
+ 'English': 'en',
9
+ 'French': 'fr',
10
+ 'German': 'de',
11
+ 'Spanish': 'es',
12
+ 'Italian': 'it',
13
+ 'Dutch': 'nl',
14
+ 'Portuguese': 'pt',
15
+ 'Russian': 'ru',
16
+ 'Chinese': 'zh',
17
+ 'Japanese': 'ja',
18
+ 'Korean': 'ko',
19
+ 'Arabic': 'ar'
20
+ }
21
+ def language_translator(source, target,text):
22
+ task = f'translation_{language_codes[source]}_to_{language_codes[target]}'
23
+
24
+ translator = pipeline(task, model="google-t5/t5-small")
25
+ translation = translator(text)
26
+
27
+ return translation[0]['translation_text']
28
+
29
+ gr.close_all()
30
+
31
+
32
+ demo = gr.Interface(language_translator,
33
+ inputs=[gr.Dropdown(['English','French','German','Spanish','Italian','Dutch','Portuguese','Russian','Chinese','Japanese','Korean','Arabic'],
34
+ label='Source Language', value='English' ),
35
+ gr.Dropdown(['English','French','German','Spanish','Italian','Dutch','Portuguese','Russian','Chinese','Japanese','Korean','Arabic'],
36
+ label='Target Language', value='French' ),
37
+ gr.Textbox(label='Text to Translate', lines=5)],
38
+
39
+ outputs=[gr.Textbox(label="Translated Text", lines=5)],
40
+ title="Gen AI Learning Project 4: Language Translator",
41
+ description="This Application uses 'google-t5/t5-small' llm to provide translation service")
42
+ demo.launch()