Spaces:
Runtime error
Runtime error
update
Browse files
app.py
CHANGED
@@ -11,30 +11,62 @@ import numpy as np
|
|
11 |
import gradio as gr
|
12 |
import numpy as np
|
13 |
|
14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
def choose_language(language):
|
16 |
-
#
|
17 |
-
#Support for Arabic, Turish, arabic
|
18 |
-
#Support for German
|
19 |
if language == 'hindi':
|
20 |
-
|
|
|
21 |
elif language == 'spanish':
|
22 |
-
|
|
|
23 |
elif language == 'german':
|
24 |
-
|
|
|
25 |
elif language == 'french':
|
26 |
-
|
|
|
27 |
elif language == 'turkish':
|
28 |
-
|
|
|
29 |
elif language == 'arabic':
|
30 |
-
|
|
|
31 |
else:
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
tokenizer = AutoTokenizer.from_pretrained(modelnm)
|
36 |
-
model = AutoModelForSeq2SeqLM.from_pretrained(modelnm)
|
37 |
-
|
38 |
return tokenizer, model
|
39 |
|
40 |
#Function to translate english text to desired language
|
|
|
11 |
import gradio as gr
|
12 |
import numpy as np
|
13 |
|
14 |
+
|
15 |
+
#Support for Hindi, Spanish, French, Arabic, Turish, Gailec/Irish, and German
|
16 |
+
|
17 |
+
#'hindi':
|
18 |
+
tokenizerhi = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-hi")
|
19 |
+
modelhi = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-hi")
|
20 |
+
|
21 |
+
#'spanish':
|
22 |
+
tokenizeres = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-es")
|
23 |
+
modeles = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-es")
|
24 |
+
|
25 |
+
#'german':
|
26 |
+
tokenizerde = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-de")
|
27 |
+
modelde = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-de")
|
28 |
+
|
29 |
+
#'french':
|
30 |
+
tokenizerfr = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
|
31 |
+
modelfr = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
|
32 |
+
|
33 |
+
#'turkish':
|
34 |
+
tokenizertrk = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-trk")
|
35 |
+
modeltrk = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-trk")
|
36 |
+
|
37 |
+
#'arabic':
|
38 |
+
tokenizerar = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
|
39 |
+
modelar = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
|
40 |
+
|
41 |
+
#Irish /Gaelish
|
42 |
+
tokenizerga = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ga")
|
43 |
+
modelga = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-ga")
|
44 |
+
|
45 |
+
|
46 |
+
#Translate in your desired language
|
47 |
def choose_language(language):
|
48 |
+
#Loading the tokenizers and trained models
|
|
|
|
|
49 |
if language == 'hindi':
|
50 |
+
tokenizer, model = tokenizerhi, modelhi
|
51 |
+
#modelnm = "Helsinki-NLP/opus-mt-en-hi"
|
52 |
elif language == 'spanish':
|
53 |
+
tokenizer, model = tokenizeres, modeles
|
54 |
+
#modelnm = "Helsinki-NLP/opus-mt-en-es"
|
55 |
elif language == 'german':
|
56 |
+
tokenizer, model = tokenizerde, modelde
|
57 |
+
#modelnm = "Helsinki-NLP/opus-mt-en-de"
|
58 |
elif language == 'french':
|
59 |
+
tokenizer, model = tokenizerfr, modelfr
|
60 |
+
#modelnm = "Helsinki-NLP/opus-mt-en-fr"
|
61 |
elif language == 'turkish':
|
62 |
+
tokenizer, model = tokenizertrk, modeltrk
|
63 |
+
#modelnm = "Helsinki-NLP/opus-mt-en-trk"
|
64 |
elif language == 'arabic':
|
65 |
+
tokenizer, model = tokenizerar, modelar
|
66 |
+
#modelnm = "Helsinki-NLP/opus-mt-en-ar"
|
67 |
else:
|
68 |
+
tokenizer, model = tokenizerga, modelga
|
69 |
+
#modelnm = "Helsinki-NLP/opus-mt-en-ga"
|
|
|
|
|
|
|
|
|
70 |
return tokenizer, model
|
71 |
|
72 |
#Function to translate english text to desired language
|