Update app.py

app.py

@@ -43,16 +43,21 @@ geolocator = Nominatim(user_agent="skin-dashboard", timeout = 10)
 
 @st.cache_resource
 def load_image_model(token: str):
-    """
-    Load the pre-trained skin cancer classification model using PyTorch.
-    """
-    try:
-        extractor = AutoFeatureExtractor.from_pretrained(MODEL_NAME)
-        model = AutoModelForImageClassification.from_pretrained(MODEL_NAME)
-        return pipeline("image-classification", model=model, feature_extractor=extractor, framework="pt")
-    except Exception as e:
-        st.error(f"Error loading the model: {e}")
-        return None
+    return pipeline(
+        "image-classification",
+        feature_extractor=AutoFeatureExtractor.from_pretrained(
+            MODEL_NAME,
+            subfolder="Skin_Cancer-Image_Classification",
+            use_auth_token=token
+        ),
+        model=AutoModelForImageClassification.from_pretrained(
+            MODEL_NAME,
+            subfolder="Skin_Cancer-Image_Classification",
+            use_auth_token=token
+        ),
+        device=0  # or -1 for CPU
+    )
+
 
 
 @st.cache_resource