fix model load

src/prediction.py

@@ -21,9 +21,8 @@ class_names = [
 
 @st.cache_resource
 def load_model_from_hub():
-    model_dir = snapshot_download("Heizsenberg/tomato-leaf-classification")
+    model = tf.keras.models.load_model("leaf_detection_model.keras")
 
-    model = tf.keras.models.load_model(model_dir)
     return model