Gurinder commited on
Commit
cac34b5
·
1 Parent(s): 1d81dc7

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +41 -0
app.py ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !pip install transformers sentencepiece -q
2
+
3
+ !pip install gradio
4
+
5
+ # Importing the pipeline function from the transformers library
6
+ from transformers import pipeline
7
+ import gradio as gr
8
+
9
+ # Creating a Text2TextGenerationPipeline for language translation
10
+ pipe = pipeline(task='text2text-generation', model='facebook/m2m100_418M')
11
+
12
+ # Define the language dictionary outside the translate function
13
+ languages = {
14
+ "English": "en",
15
+ "Spanish": "es",
16
+ "French": "fr",
17
+ "German": "de",
18
+ "Chinese": "zh",
19
+ "Hindi": "hi"
20
+ }
21
+
22
+ def translate(text, target_language):
23
+ target_lang_id = pipe.tokenizer.get_lang_id(lang=languages[target_language])
24
+ translated_text = pipe(text, forced_bos_token_id=target_lang_id)
25
+ return translated_text[0]['generated_text']
26
+ gr.close_all()
27
+
28
+ iface = gr.Interface(
29
+ fn=translate,
30
+ title="Text Translator",
31
+ inputs=[
32
+ gr.inputs.Textbox(lines=2, label="Input Text"),
33
+ gr.inputs.Dropdown(list(languages.keys()), label="Target Language")
34
+ ],
35
+ outputs=gr.outputs.Textbox(label="Translated Text")
36
+ )
37
+
38
+ iface.launch()
39
+
40
+
41
+