Update app.py

app.py

@@ -4,20 +4,10 @@ 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
+import os
 
-# Define custom optimizer for loading saved model
-class CustomAdam(optimizers.Adam):
-    def get_config(self):
-        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})
+# Load pre-trained model
+model = tf.keras.models.load_model('facial_condition_model.h5')
 
 # Function to preprocess image for model input
 def preprocess_image(image):