Modify requirements.txt and some fix at app

app.py

@@ -9,15 +9,28 @@ translation_pipelines = {
     "I": pipeline("translation_en_to_it", model="Helsinki-NLP/opus-mt-en-it")
 }
 
+# Mapear los valores de entrada de Gradio a las claves del diccionario
+language_map = {
+    "German": "G",
+    "French": "F",
+    "Spanish": "S",
+    "Italian": "I"
+}
+
 def translate(text, target_language):
     if not target_language:
         return "Please select a target language."
 
-    pipe = translation_pipelines[target_language[0]]
+    # Mapeamos el idioma a la clave correspondiente en el diccionario
+    target_language_code = language_map.get(target_language, None)
+
+    if not target_language_code:
+        raise ValueError(f"Language '{target_language}' not supported")
 
-    if not pipe:
-        raise ValueError(f"Language '{target_language[0]}' not supported")
+    # Usar el pipeline correspondiente
+    pipe = translation_pipelines[target_language_code]
 
+    # Realizar la traducción
     translation = pipe(text)
     translated_text = translation[0]['translation_text']
     return translated_text

requirements.txt

@@ -1,3 +1,4 @@
 torch
 transformers
-sentencepiece
+sentencepiece
+sacremoses