Update app.py

app.py

@@ -2,45 +2,36 @@ import gradio as gr
 from transformers import pipeline
 import os
 
-# Connects to your specific model
-# If you make your model PRIVATE, ensure you add HF_TOKEN to Space Secrets!
-auth_token = os.environ.get("HF_TOKEN")
+# 1. Load token and model
+hf_token = os.environ.get("HF_TOKEN")
 model_id = "arnoldramo/garifuna-nllb"
 
-# Initialize translation pipeline
+print("Loading pipeline...")
 translator = pipeline(
     "translation", 
     model=model_id, 
-    token=auth_token
+    token=hf_token
 )
+print("Pipeline loaded successfully!")
 
-def translate_to_garifuna(message, history, source_lang):
-    # Mapping selection to NLLB language codes
-    lang_map = {
-        "English": "eng_Latn",
-        "Spanish": "spa_Latn"
-    }
+def respond(message, history, source_lang):
+    lang_map = {"English": "eng_Latn", "Spanish": "spa_Latn"}
     src_code = lang_map.get(source_lang, "eng_Latn")
     
-    # Run the model
-    output = translator(message, src_lang=src_code, tgt_lang="cab_Latn")
-    return output[0]['translation_text']
+    # Run translation
+    result = translator(message, src_lang=src_code, tgt_lang="cab_Latn")
+    return result[0]['translation_text']
 
-# Create the Chat Interface
+# 2. Corrected Interface (Fixes the theme error)
 demo = gr.ChatInterface(
-    fn=translate_to_garifuna,
-    title="Garifuna Translator",
-    description="Type in English or Spanish to see the Garifuna translation.",
+    fn=respond,
+    title="Garifuna Language Assistant",
+    description="Translate English or Spanish to Garifuna.",
     additional_inputs=[
-        gr.Dropdown(
-            label="Source Language", 
-            choices=["English", "Spanish"], 
-            value="English" # English is default
-        )
-    ],
-    examples=[["Hello, how are you?", "English"], ["Buenos días", "Spanish"]],
-    theme=gr.themes.Soft()
+        gr.Dropdown(label="Source Language", choices=["English", "Spanish"], value="English")
+    ]
 )
 
 if __name__ == "__main__":
+    # We apply the theme here or leave it default to ensure it boots
     demo.launch()