Hugging Face's logo

Spaces:
yuragoithf
/
mlg_image_classification
Sleeping

App Files Community
yuragoithf commited on
Commit
773cb82
·
verified ·
1 Parent(s): 05b1421

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
- import tensorflow as tf
 
3
  import gdown
4
  from PIL import Image
5
  import pillow_avif
@@ -30,7 +31,7 @@ def download_model():
30
  model_file = download_model()
31
 
32
  # Load the model
33
- model = tf.keras.models.load_model(model_file)
34
 
35
  # Perform image classification for single class output
36
  # def predict_class(image):
 
1
  import gradio as gr
2
+ 'import tensorflow as tf
3
+ import keras
4
  import gdown
5
  from PIL import Image
6
  import pillow_avif
 
31
  model_file = download_model()
32
 
33
  # Load the model
34
+ model = keras.saving.load_model(model_file)
35
 
36
  # Perform image classification for single class output
37
  # def predict_class(image):