Update app.py

app.py

@@ -78,12 +78,13 @@ def translate(input_text, source, target):
         target = target_lang_dict[target]
 
     try:
-        model_id = f"Helsinki-NLP/opus-mt-{source}-{target}"
-        pipe = pipeline("translation", model=model_id, device=-1)  # CPU
-        translation = pipe(input_text)
-        return translation[0]['translation_text']
+          target = target_lang_dict[target]
+          model = f"Helsinki-NLP/opus-mt-{source}-{target}"
+          pipe = pipeline("translation", model=model)
+          translation = pipe(input_text)
+          return translation[0]['translation_text']
     except KeyError:
-        return f"Error: Translation from {source_readable} to {target} is not supported"
+          return f"Error: Translation from {source_readable} to {target} is not supported"
 
 with st.container():            
   st.header("Basque (EU) Euskera Multilingual Translator (Basque as the Starting Point)")