Hugging Face's logo

Spaces:
yuragoithf
/
mlg_image_classification
Sleeping

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -30,8 +30,7 @@ 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
- model = tf.saved_model.load(model_file)
35
 
36
  # Perform image classification for single class output
37
  # 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
 
35
  # Perform image classification for single class output
36
  # def predict_class(image):