Create app.py

app.py

@@ -0,0 +1,46 @@
+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")
+model_id = "arnoldramo/garifuna-nllb"
+
+# Initialize translation pipeline
+translator = pipeline(
+    "translation", 
+    model=model_id, 
+    token=auth_token
+)
+
+def translate_to_garifuna(message, history, source_lang):
+    # Mapping selection to NLLB language codes
+    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']
+
+# Create the Chat Interface
+demo = gr.ChatInterface(
+    fn=translate_to_garifuna,
+    title="Garifuna Translator",
+    description="Type in English or Spanish to see the Garifuna translation.",
+    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()
+)
+
+if __name__ == "__main__":
+    demo.launch()