update

app.py

@@ -11,30 +11,62 @@ import numpy as np
 import gradio as gr
 import numpy as np
 
-#Translated in your desired language
+
+#Support for Hindi, Spanish, French, Arabic, Turish, Gailec/Irish, and German
+
+#'hindi':
+tokenizerhi = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-hi")
+modelhi = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-hi")
+
+#'spanish':
+tokenizeres = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-es")
+modeles = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-es")
+
+#'german':
+tokenizerde = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-de")
+modelde = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-de")
+
+#'french':
+tokenizerfr = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
+modelfr = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
+
+#'turkish':
+tokenizertrk = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-trk")
+modeltrk = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-trk")
+
+#'arabic':
+tokenizerar = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
+modelar = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
+
+#Irish /Gaelish
+tokenizerga = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ga")
+modelga = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-ga")
+
+
+#Translate in your desired language
 def choose_language(language):
-  #Support for Hindi, Spanish, French
-  #Support for Arabic, Turish, arabic
-  #Support for German
+  #Loading the tokenizers and trained models
   if language == 'hindi':
-    modelnm = "Helsinki-NLP/opus-mt-en-hi"
+    tokenizer, model = tokenizerhi, modelhi
+    #modelnm = "Helsinki-NLP/opus-mt-en-hi"
   elif language == 'spanish':
-    modelnm = "Helsinki-NLP/opus-mt-en-es"
+    tokenizer, model = tokenizeres, modeles
+    #modelnm = "Helsinki-NLP/opus-mt-en-es"
   elif language == 'german':
-    modelnm = "Helsinki-NLP/opus-mt-en-de"
+    tokenizer, model = tokenizerde, modelde
+    #modelnm = "Helsinki-NLP/opus-mt-en-de"
   elif language == 'french':
-    modelnm = "Helsinki-NLP/opus-mt-en-fr"
+    tokenizer, model = tokenizerfr, modelfr
+    #modelnm = "Helsinki-NLP/opus-mt-en-fr"
   elif language == 'turkish':
-    modelnm = "Helsinki-NLP/opus-mt-en-trk"
+    tokenizer, model = tokenizertrk, modeltrk
+    #modelnm = "Helsinki-NLP/opus-mt-en-trk"
   elif language == 'arabic':
-    modelnm = "Helsinki-NLP/opus-mt-en-ar"
+    tokenizer, model = tokenizerar, modelar
+    #modelnm = "Helsinki-NLP/opus-mt-en-ar"
   else:
-    modelnm = "Helsinki-NLP/opus-mt-en-ga"
-
-
-  tokenizer = AutoTokenizer.from_pretrained(modelnm)
-  model = AutoModelForSeq2SeqLM.from_pretrained(modelnm)
-
+    tokenizer, model = tokenizerga, modelga
+    #modelnm = "Helsinki-NLP/opus-mt-en-ga"
   return tokenizer, model
 
 #Function to translate english text to desired language