Update app.py

app.py

@@ -3,20 +3,20 @@ from huggingface_hub import InferenceClient, HfApi
 from huggingface_hub.utils import HfHubHTTPError
 import os
 
-def check_api_status(model_id):
+def check_api_status(model_id, token):
     try:
-        client = InferenceClient(model_id)
-        # Simple test request to check if API is responsive
-        client.health_check()
-        return "API is accessible"
+        client = InferenceClient(model_id, token=token)
+        # Test if we can connect to the API
+        response = client.chat_completion(
+            [{"role": "user", "content": "test"}],
+            max_tokens=1,
+            stream=False
+        )
+        return "API is accessible and responding"
     except Exception as e:
-        try:
-            # Try to enable the API
-            api = HfApi()
-            api.toggle_model_api(model_id, True)
-            return "API was disabled. Now enabled for the model."
-        except Exception as enable_err:
-            return f"Could not enable API: {str(enable_err)}"
+        if "rate limit" in str(e).lower():
+            return "API is accessible (rate limited)"
+        return f"API status: {str(e)}"
 
 def get_api_status():
     token = os.getenv('HF_TOKEN')
@@ -26,8 +26,7 @@ def get_api_status():
         return "⚠️ No API token found. Please set HF_TOKEN environment variable."
     
     try:
-        client = InferenceClient(model_id, token=token)
-        status = check_api_status(model_id)
+        status = check_api_status(model_id, token)
         return f"✅ Connected to {model_id} | {status}"
     except Exception as e:
         return f"❌ Error: {str(e)}"
@@ -107,7 +106,7 @@ with gr.Blocks() as demo:
         ],
     )
     
-    # Add API status at the footer with error handling
+    # Add API status at the footer with improved status check
     footer = gr.HTML(
         value=f"<div style='text-align: center; padding: 10px; background-color: #f0f0f0; border-top: 1px solid #ddd;'>{get_api_status()}</div>",
         every=30  # Updates every 30 seconds