Add authentication support for private model access

- Add HF_TOKEN support for accessing gated/private model repository
- Update README.md with setup instructions for authentication
- Add detailed error handling for authentication issues
- Provide clear instructions for adding HF_TOKEN to Space secrets

README.md

@@ -11,4 +11,24 @@ license: mit
 short_description: Derma AI skin Disease model
 ---
 
+# DermaAI - Skin Disease Classification
+
+AI-powered skin condition analysis using deep learning with EfficientNetV2 architecture.
+
+## Setup
+
+This Space requires authentication to access the private model. To run this Space:
+
+1. Go to the Space settings
+2. Add a new secret with key `HF_TOKEN` and your Hugging Face access token as the value
+3. Make sure your token has access to the `Siraja704/DermaAI` model repository
+
+## Supported Conditions
+
+- Atopic Dermatitis
+- Eczema
+- Psoriasis
+- Seborrheic Keratoses
+- Tinea Ringworm Candidiasis
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -33,14 +33,25 @@ class DermaAIModel:
         """Load the DermaAI model from Hugging Face Hub"""
         try:
             print("🔄 Loading DermaAI model from Hugging Face...")
+            # Get HF token from environment variable for authentication
+            hf_token = os.getenv("HF_TOKEN")
             model_path = hf_hub_download(
                 repo_id="Siraja704/DermaAI", 
-                filename="DermaAI.keras"
+                filename="DermaAI.keras",
+                token=hf_token
             )
             self.model = tf.keras.models.load_model(model_path)
             print("✅ Model loaded successfully!")
         except Exception as e:
+            error_msg = str(e)
             print(f"❌ Error loading model: {e}")
+            if "401" in error_msg or "gated" in error_msg.lower() or "restricted" in error_msg.lower():
+                print("\n🔐 AUTHENTICATION ERROR:")
+                print("- The model repository is private/gated")
+                print("- Please add your HF_TOKEN to the Space secrets")
+                print("- Go to Space Settings > Repository secrets")
+                print("- Add: HF_TOKEN = your_huggingface_token")
+                print("- Make sure the token has access to Siraja704/DermaAI\n")
             raise e
     
     def predict(self, image):