Update app/model_loader.py

app/model_loader.py

@@ -1,24 +1,8 @@
-# app/model_loader.py
 import os
 import tensorflow as tf
-from huggingface_hub import hf_hub_download
 
-# Use /data because Hugging Face Spaces guarantee it's writable
-MODEL_DIR = "/data/model"
-MODEL_PATH = os.path.join(MODEL_DIR, "efficientnetv2s.h5")
-REPO_ID = "Miguel764/efficientnetv2s-skin-cancer-classifier"
-FILENAME = "efficientnetv2s.h5"
+MODEL_PATH = "app/model/efficientnetv2s.h5"
 
 def load_model():
-    if not os.path.exists(MODEL_PATH):
-        print("Model not found locally. Downloading from Hugging Face...")
-        os.makedirs(MODEL_DIR, exist_ok=True)
-        hf_hub_download(
-            repo_id=REPO_ID,
-            filename=FILENAME,
-            local_dir=MODEL_DIR
-        )
-    else:
-        print("Model already exists locally.")
-
+    print("Loading pre-downloaded model...")
     return tf.keras.models.load_model(MODEL_PATH)