Update app.py

app.py

@@ -8,17 +8,21 @@ from huggingface_hub.utils import RepositoryNotFoundError, HfHubHTTPError
 def initialize_model():
     try:
         # Login to Hugging Face Hub
-        login(token=os.getenv("HUGGING_FACE_HUB_TOKEN"))
+        token = os.getenv("HUGGING_FACE_HUB_TOKEN")
+        if not token:
+            return False, "No token found. Please set HUGGING_FACE_HUB_TOKEN in Space secrets.", None
+            
+        login(token=token)
         
-        # Initialize the model and tokenizer with auth
+        # Initialize the model and tokenizer with token (not use_auth_token)
         model_id = "CohereLabs/c4ai-command-a-03-2025"
         tokenizer = AutoTokenizer.from_pretrained(
             model_id,
-            use_auth_token=True
+            token=token  # Updated from use_auth_token
         )
         model = AutoModelForCausalLM.from_pretrained(
             model_id,
-            use_auth_token=True
+            token=token  # Updated from use_auth_token
         )
         return True, model, tokenizer
     except RepositoryNotFoundError:
@@ -27,7 +31,7 @@ def initialize_model():
         if e.response.status_code == 401:
             return False, "Authentication failed. Please check your token permissions.", None
         elif e.response.status_code == 403:
-            return False, "Access denied. Please ensure you have access to this model.", None
+            return False, "Access denied. Please request access at https://huggingface.co/CohereLabs/c4ai-command-a-03-2025", None
         else:
             return False, f"An error occurred: {str(e)}", None
     except Exception as e:
@@ -37,7 +41,6 @@ def initialize_model():
 success, result, tokenizer = initialize_model()
 if not success:
     print(f"Error initializing model: {result}")
-    # You might want to raise an exception here or handle the error differently
 else:
     model = result
 
@@ -85,7 +88,7 @@ def chat(message, history):
     except Exception as e:
         return [(message, f"Error during chat: {str(e)}")]
 
-# Create the Gradio interface
+# Create the Gradio interface - removed retry_on_error
 demo = gr.ChatInterface(
     fn=chat,
     title="Medical Decision Support AI",
@@ -95,8 +98,7 @@ demo = gr.ChatInterface(
         "What are the symptoms of hypertension?",
         "What are common drug interactions with aspirin?",
         "What are the warning signs of diabetes?",
-    ],
-    retry_on_error=True
+    ]
 )
 
 demo.launch()