Update app.py

app.py

@@ -47,8 +47,8 @@ def load_image_model(token: str):
     Load the pre-trained skin cancer classification model using PyTorch.
     """
     try:
-        extractor = AutoFeatureExtractor.from_pretrained("Anwarkh1/Skin_Cancer-Image_Classification")
-        model = AutoModelForImageClassification.from_pretrained("Anwarkh1/Skin_Cancer-Image_Classification")
+        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}")