Update app.py

app.py

@@ -4,6 +4,8 @@ import urllib
 import tensorflow as tf
 from tensorflow.keras import layers, models, optimizers
 import gradio as gr
+from tensorflow.keras.utils import get_custom_objects
+from app import CustomAdam
 
 # Define custom optimizer for loading saved model
 class CustomAdam(optimizers.Adam):
@@ -11,6 +13,8 @@ class CustomAdam(optimizers.Adam):
         config = super().get_config()
         config['name'] = 'CustomAdam'
         return config
+        
+get_custom_objects().update({'CustomAdam': CustomAdam})
 
 # Load pre-trained model with custom optimizer
 model = tf.keras.models.load_model('facial_condition_model.h5', custom_objects={'CustomAdam': CustomAdam})