Update app.py

app.py

@@ -5,8 +5,15 @@ import tensorflow as tf
 from tensorflow.keras import layers, models, optimizers
 import gradio as gr
 
-# Load pre-trained model
-model = tf.keras.models.load_model('facial_condition_model.h5')
+# Define custom optimizer for loading saved model
+class CustomAdam(optimizers.Adam):
+    def get_config(self):
+        config = super().get_config()
+        config['name'] = 'CustomAdam'
+        return config
+
+# Load pre-trained model with custom optimizer
+model = tf.keras.models.load_model('facial_condition_model.h5', custom_objects={'CustomAdam': CustomAdam})
 
 # Function to preprocess image for model input
 def preprocess_image(image):