Update app.py

app.py

@@ -43,32 +43,31 @@ geolocator = Nominatim(user_agent="skin-dashboard", timeout = 10)
 
 @st.cache_resource
 def load_image_model(token: str):
-    # 1) load the feature extractor from the Hub as usual
+    # 1) Grab a working ConvNeXt feature extractor from the official base model
     extractor = AutoFeatureExtractor.from_pretrained(
-        MODEL_NAME,
-        use_auth_token=token
+        "microsoft/convnext-base-224"
     )
 
-    # 2) manually create a ConvNextConfig with the right num_labels / id2label
+    # 2) Build a minimal ConvNextConfig that matches your fine-tuned head
     config = ConvNextConfig(
         num_labels=2,
         id2label={0: "benign", 1: "malignant"},
-        label2id={"benign": 0, "malignant": 1}
+        label2id={"benign": 0, "malignant": 1},
     )
 
-    # 3) load the weights with that config override
+    # 3) Load *just the weights* from your repo, using the config override
     model = AutoModelForImageClassification.from_pretrained(
         MODEL_NAME,
         config=config,
-        use_auth_token=token
+        use_auth_token=token,
     )
 
-    # 4) build your pipeline
+    # 4) Return the pipeline
     return pipeline(
         "image-classification",
         model=model,
         feature_extractor=extractor,
-        device=0  # or -1 for CPU
+        device=0,  # or -1 for CPU
     )
 
 @st.cache_resource