Update app.py

app.py

@@ -39,12 +39,11 @@ def load_model():
         # Try loading with different configurations
         try:
             logger.info("Attempting to load model with custom objects...")
-            with tf.keras.utils.custom_object_scope(custom_objects):
-                model = tf_keras.models.load_model(model_path, compile=False)
+            model = tf_keras.models.load_model(model_path, custom_objects={'KerasLayer': hub.KerasLayer})
         except Exception as e:
             logger.error(f"Failed to load with custom objects: {str(e)}")
             logger.info("Attempting to load model without custom objects...")
-            model = tf_keras.models.load_model(model_path, compile=False)
+            model = tf_keras.models.load_model(model_path)
         
         # Verify model loaded correctly
         if model is None: