Update app.py

app.py

@@ -50,7 +50,8 @@ def predict_image(img):
 
 # Charger le modèle pré-entraîné
 # Assurez-vous que le chemin du modèle et de l'encodeur sont corrects
-model = tf.keras.models.load_model('trained_model.h5')
+model = tf.keras.models.load_model('trained_model.h5', custom_objects={'mse': MeanSquaredError()})
+
 label_encoder = LabelEncoder()
 label_encoder.classes_ = np.load('path_to_label_encoder_classes.npy')