Spaces:

yuragoithf
/

mlg_image_classification

Sleeping

yuragoithf commited on Sep 30, 2024

Commit

eb4128b

verified ·

1 Parent(s): 014e0c0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ def download_model():
 model_file = download_model()
 # Load the model
-model = tf.keras.models.load_model(model_file)
 # Perform image classification for single class output
 # def predict_class(image):

 model_file = download_model()
 # Load the model
+model = tf.keras.models.load_model(model_file, compile=True)
 # Perform image classification for single class output
 # def predict_class(image):