Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
+import os
 from huggingface_hub import InferenceClient
 
-# Initialize the Hugging Face Inference Client
-# No API key needed for public models, but you can add one for better rate limits
-client = InferenceClient()
+# Initialize the Hugging Face Inference Client with token from environment
+# Get token from HuggingFace Spaces secrets or environment variable
+hf_token = os.environ.get("HF_TOKEN", None)
+client = InferenceClient(token=hf_token)
 
 def chatbot(question, model, temperature):
     """Send the question to HuggingFace model using the Inference API.
@@ -29,7 +31,30 @@ def chatbot(question, model, temperature):
         return f"Model: {model}\nTemperature: {temperature}\n\n{content}"
     
     except Exception as e:
-        return f"Error: {str(e)}\n\nPlease try again or select a different model."
+        error_msg = str(e)
+        if "api_key" in error_msg.lower() or "authentication" in error_msg.lower():
+            return """⚠️ Authentication Error: This model requires a HuggingFace token.
+
+Please follow these steps:
+
+1. Get a FREE token from: https://huggingface.co/settings/tokens
+   - Click "New token"
+   - Name it (e.g., "chatbot-token")
+   - Select "Read" permission
+   - Click "Create token" and copy it
+
+2. Add the token to your Space:
+   - Go to your Space Settings → "Variables and secrets"
+   - Click "New secret"
+   - Name: HF_TOKEN
+   - Value: [paste your token]
+   - Save
+
+3. Restart your Space (it will rebuild automatically)
+
+OR try selecting a different model that doesn't require authentication."""
+        else:
+            return f"Error: {error_msg}\n\nPlease try again or select a different model."
 
 
 def main():
@@ -38,6 +63,12 @@ def main():
         gr.Markdown("# 🤗 HuggingFace Chatbot")
         gr.Markdown("Ask questions and get responses from various HuggingFace AI models!")
         
+        # Show token status
+        if hf_token:
+            gr.Markdown("✅ **Status**: HuggingFace token configured")
+        else:
+            gr.Markdown("⚠️ **Status**: No HuggingFace token found. Some models may not work. See instructions below if you get errors.")
+        
         with gr.Row():
             # Left column: grouped controls (model dropdown + temperature slider)
             with gr.Column(scale=1):
@@ -45,12 +76,14 @@ def main():
                 model_dropdown = gr.Dropdown(
                     label="Model",
                     choices=[
+                        "meta-llama/Llama-3.2-1B-Instruct",
                         "meta-llama/Llama-3.2-3B-Instruct",
                         "microsoft/Phi-3-mini-4k-instruct",
+                        "HuggingFaceH4/zephyr-7b-beta",
                         "mistralai/Mistral-7B-Instruct-v0.3",
-                        "HuggingFaceH4/zephyr-7b-beta"
+                        "google/gemma-2-2b-it"
                     ],
-                    value="meta-llama/Llama-3.2-3B-Instruct",
+                    value="meta-llama/Llama-3.2-1B-Instruct",
                 )
                 temp_slider = gr.Slider(
                     label="Temperature",
@@ -90,4 +123,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-