Update app.py

app.py

@@ -1,8 +1,30 @@
 import gradio as gr
 from transformers import pipeline
+from huggingface_hub import login
+import os
 
-# Load the speech-to-text model
-transcriber = pipeline("automatic-speech-recognition", model="NCAIR1/NigerianAccentedEnglish")
+# Note: This model (NCAIR1/NigerianAccentedEnglish) is a gated model
+# For demo purposes, we're using openai/whisper-small as a fallback
+# To use the NCAIR model, you need to:
+# 1. Accept the model terms at: https://huggingface.co/NCAIR1/NigerianAccentedEnglish
+# 2. Add your HF token as a secret named HF_TOKEN in Space settings
+
+try:
+    # Try to use authentication if available
+    token = os.getenv("HF_TOKEN")
+    if token:
+        login(token=token)
+        transcriber = pipeline("automatic-speech-recognition", model="NCAIR1/NigerianAccentedEnglish")
+        model_name = "NCAIR1/NigerianAccentedEnglish (Nigerian Accented English)"
+    else:
+        # Fallback to open model
+        transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-small")
+        model_name = "OpenAI Whisper Small (General English)"
+except Exception as e:
+    # Fallback to open model if gated model fails
+    print(f"Error loading NCAIR model: {e}")
+    transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-small")
+    model_name = "OpenAI Whisper Small (General English - Fallback)"
 
 def transcribe_audio(audio):
     if audio is None:
@@ -17,8 +39,8 @@ demo = gr.Interface(
     fn=transcribe_audio,
     inputs=gr.Audio(type="filepath", label="Upload Audio"),
     outputs=gr.Textbox(label="Transcript"),
-    title="Nigerian Accented English Speech-to-Text",
-    description="Upload an audio file to transcribe Nigerian accented English speech using the NCAIR1/NigerianAccentedEnglish model."
+    title="Nigerian Accented English Speech-to-Text Demo",
+    description=f"Upload an audio file to transcribe speech to text.\n\nCurrently using: **{model_name}**\n\nNote: To use the NCAIR1/NigerianAccentedEnglish model, accept terms at https://huggingface.co/NCAIR1/NigerianAccentedEnglish and add your HF token."
 )
 
 if __name__ == "__main__":