Spaces:

Fluospark128
/

Cassava_Disease_Classifier

Running

Fluospark128 commited on Feb 10

Commit

3659d89

verified ·

1 Parent(s): 776d17a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ from PIL import Image
 # Load the model
 @st.cache_resource
 def load_model():
-    model = tf.keras.models.load_model("hf://Fluospark128/Cassava_Disease_Classifier/cassava_model.keras") #tf.keras.saving.load_model("hf://Fluospark128/Cassava_Disease_Classifier") cassava_model.h5
     return model
 model = load_model()

 # Load the model
 @st.cache_resource
 def load_model():
+    model = tf.keras.models.load_model("cassava_leaf_disease_model.keras") #tf.keras.saving.load_model("hf://Fluospark128/Cassava_Disease_Classifier") cassava_model.h5
     return model
 model = load_model()