Hugging Face's logo

Spaces:
yuragoithf
/
mlg_image_classification
Sleeping

App Files Community
yuragoithf commited on
Commit
83e0917
·
verified ·
1 Parent(s): 9d3cbd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -30,7 +30,8 @@ def download_model():
30
  model_file = download_model()
31
 
32
  # Load the model
33
- model = tf.keras.models.load_model(model_file, compile=True)
 
34
 
35
  # Perform image classification for single class output
36
  # def predict_class(image):
 
30
  model_file = download_model()
31
 
32
  # Load the model
33
+ # model = tf.keras.models.load_model(model_file, compile=True)
34
+ model = tf.saved_model.load(model_file)
35
 
36
  # Perform image classification for single class output
37
  # def predict_class(image):