saaara commited on
Commit
0d6ba95
·
verified ·
1 Parent(s): 1a04da2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,8 +52,8 @@ def predict_image(img):
52
  # Assurez-vous que le chemin du modèle et de l'encodeur sont corrects
53
  model = tf.keras.models.load_model('trained_model.h5', custom_objects={'mse': MeanSquaredError()})
54
 
55
- label_encoder = LabelEncoder()
56
- label_encoder.classes_ = np.load('path_to_label_encoder_classes.npy')
57
 
58
  # Interface Gradio
59
  interface = gr.Interface(
 
52
  # Assurez-vous que le chemin du modèle et de l'encodeur sont corrects
53
  model = tf.keras.models.load_model('trained_model.h5', custom_objects={'mse': MeanSquaredError()})
54
 
55
+ #label_encoder = LabelEncoder()
56
+ #label_encoder.classes_ = np.load('path_to_label_encoder_classes.npy')
57
 
58
  # Interface Gradio
59
  interface = gr.Interface(