Update app.py

app.py

@@ -8,9 +8,14 @@ HF_TOKEN = os.environ.get("HF_TOKEN", "")
 
 def query_model(prompt, max_tokens=256, temperature=0.7):
     """Query the model using Hugging Face Inference API"""
+    
+    # Check if token is available
+    if not HF_TOKEN:
+        return "❌ Error: Hugging Face token not found. Please add HF_TOKEN in Space settings → Repository secrets."
+    
     try:
         API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
-        headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
+        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
         
         payload = {
             "inputs": prompt,
@@ -30,18 +35,26 @@ def query_model(prompt, max_tokens=256, temperature=0.7):
                 return result[0].get('generated_text', 'No generated text found')
             else:
                 return f"Unexpected response format: {result}"
+        elif response.status_code == 401:
+            return "❌ Error: Invalid Hugging Face token. Please check your HF_TOKEN in Space settings."
+        elif response.status_code == 503:
+            # Model is loading
+            return "🔄 Model is loading, please wait 20-30 seconds and try again..."
         else:
             error_msg = response.json().get('error', 'Unknown error')
-            return f"API Error ({response.status_code}): {error_msg}"
+            return f"❌ API Error ({response.status_code}): {error_msg}"
             
     except Exception as e:
-        return f"Request failed: {str(e)}"
+        return f"❌ Request failed: {str(e)}"
 
 # Create Gradio interface
-with gr.Blocks(title="0dAI 8x7B Model Demo") as demo:
+with gr.Blocks(title="0dAI 8x7B Model Demo", theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
-    # 0dAI 8x7B Model Demo
-    This Space uses the Hugging Face Inference API to run the 0dAI/0dAI-8x7b-0761 model.
+    # 🤖 0dAI 8x7B Model Demo
+    
+    This Space uses the Hugging Face Inference API to run the [0dAI/0dAI-8x7b-0761](https://huggingface.co/0dAI/0dAI-8x7b-0761) model.
+    
+    **Note:** Make sure you've added your Hugging Face token in Space settings!
     """)
     
     with gr.Row():
@@ -69,13 +82,13 @@ with gr.Blocks(title="0dAI 8x7B Model Demo") as demo:
                     label="Temperature"
                 )
             
-            generate_btn = gr.Button("Generate", variant="primary")
+            generate_btn = gr.Button("🚀 Generate", variant="primary")
         
         with gr.Column():
             output = gr.Textbox(
                 label="Generated Text",
                 lines=8,
-                interactive=False
+                show_copy_button=True
             )
     
     # Examples
@@ -84,11 +97,21 @@ with gr.Blocks(title="0dAI 8x7B Model Demo") as demo:
             ["Explain quantum computing in simple terms:"],
             ["Write a short story about a robot learning to paint:"],
             ["What are the benefits of renewable energy?"],
-            ["How does machine learning work?"]
+            ["How does machine learning work?"],
+            ["Write a Python function to calculate fibonacci sequence:"]
         ],
-        inputs=prompt_input
+        inputs=prompt_input,
+        label="Click any example to try:"
     )
     
+    # Status information
+    gr.Markdown("""
+    ### 🔧 Setup Instructions:
+    1. Get your Hugging Face token from [Settings → Tokens](https://huggingface.co/settings/tokens)
+    2. Add it to this Space in **Settings → Repository secrets** as `HF_TOKEN`
+    3. The model may take 20-30 seconds to load on first request
+    """)
+    
     generate_btn.click(
         fn=query_model,
         inputs=[prompt_input, max_tokens, temperature],