Update app.py

app.py

@@ -44,21 +44,12 @@ geolocator = Nominatim(user_agent="skin-dashboard", timeout = 10)
 # --- Load Model & Feature Extractor ---
 @st.cache_resource
 def load_image_model(token: str):
-    processor = AutoImageProcessor.from_pretrained(
-        MODEL_NAME,
-        use_auth_token=token,
-        trust_remote_code=True
-    )
-    model = AutoModelForImageClassification.from_pretrained(
-        MODEL_NAME,
-        use_auth_token=token,
-        trust_remote_code=True
-    )
     return pipeline(
-        "image-classification",
-        model=model,
-        image_processor=processor,
-        device=0
+    "image-classification",
+    MODEL_NAME,
+    use_auth_token=token,
+    trust_remote_code=True,  # ← allow custom code in the model repo
+    device=0                 # or -1 for CPU
     )
 
 @st.cache_resource