fix missing model

src/prediction.py

@@ -2,7 +2,7 @@ import numpy as np
 import streamlit as st
 import tensorflow as tf
 from PIL import Image
-from huggingface_hub import snapshot_download
+from huggingface_hub import hf_hub_download
 import os
 
 
@@ -21,8 +21,12 @@ class_names = [
 
 @st.cache_resource
 def load_model_from_hub():
-    model = tf.keras.models.load_model("leaf_detection_model.keras")
-
+    model_path = hf_hub_download(
+        repo_id="Heizsenberg/leaf_classification",
+        filename="leaf_detection_model.keras",
+        repo_type="space"   # VERY IMPORTANT
+    )
+    model = tf.keras.models.load_model(model_path)
     return model