Update app/model_loader.py

app/model_loader.py

@@ -1,8 +1,11 @@
-import os
+from huggingface_hub import hf_hub_download
 import tensorflow as tf
 
-MODEL_PATH = "app/model/efficientnetv2s.h5"
-
 def load_model():
-    print("Loading pre-downloaded model...")
-    return tf.keras.models.load_model(MODEL_PATH)
+    # Download the model file from the public repo
+    model_path = hf_hub_download(
+        repo_id="Miguel764/efficientnetv2s-skin-cancer-classifier",  # public repo
+        filename="efficientnetv2s.h5"  # file name inside the repo
+    )
+    # Load the downloaded model
+    return tf.keras.models.load_model(model_path)