detect error fixed

app.py

@@ -1,6 +1,3 @@
-"""
-@author: idoia lerchundi
-"""
 import streamlit as st
 from language_directions import *
 from transformers import pipeline
@@ -69,19 +66,24 @@ def input_changed(source_language_dropdown, input_text=""):
 
 def translate(input_text, source, target):
     source_readable = source
+
+    # Detect source if needed
     if source == "Detect" or source.startswith("Detected"):
-      source, _ = auto_detect_language_code(input_text)
-    if source in source_lang_dict.keys():
-      source = source_lang_dict[source]
-    target_lang_dict, _ = get_target_languages(source)
+        source, _ = auto_detect_language_code(input_text)
+
+    # Map from human-readable name to code if needed
+    if source in source_lang_dict:
+        source = source_lang_dict[source]
+    if target in target_lang_dict:
+        target = target_lang_dict[target]
+
     try:
-      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']
+        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']
     except KeyError:
-      return f"Error: Translation from {source_readable} to {target} is not supported by Translation Models"
+        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)")