Spaces:

Heizsenberg
/

leaf_classification

Sleeping

Heizsenberg commited on Feb 16

Commit

79cdac4

1 Parent(s): 32b3b0a

fix model typo

Files changed (1) hide show

src/prediction.py CHANGED Viewed

@@ -21,7 +21,7 @@ class_names = [
 @st.cache_resource
 def load_model_from_hub():
     model_path = hf_hub_download(
-        repo_id="Heizsenberg/leaf-classification-tomato",
         filename="leaf_detection_model.keras"
     )
     model = tf.keras.models.load_model(model_path)

 @st.cache_resource
 def load_model_from_hub():
     model_path = hf_hub_download(
+        repo_id="Heizsenberg/tomato-leaf-classification",
         filename="leaf_detection_model.keras"
     )
     model = tf.keras.models.load_model(model_path)