Fix loading of model

app.py

@@ -1,9 +1,23 @@
 import gradio as gr
 import tensorflow as tf
 import numpy as np
+from tensorflow.keras.models import load_model 
+import keras
 
 # 1. Load your model
-model = tf.keras.models.load_model("facenet_real_fake_classifier_final.keras")
+@keras.saving.register_keras_serializable()
+def scaling(x, scale=255.0, **kwargs):
+    return x * scale
+
+@keras.saving.register_keras_serializable()
+def l2_normalize(x, axis=-1, epsilon=1e-10):
+    return tf.nn.l2_normalize(x, axis=axis, epsilon=epsilon)
+
+# Pass the registered functions explicitly
+custom_objects = {"scaling": scaling, "l2_normalize": l2_normalize}
+
+# Load the model with custom_objects
+model = load_model("facenet_real_fake_classifier_final.keras", custom_objects=custom_objects)
 
 # 2. Define your inference function
 def detect_forgery(image):

requirements.txt

@@ -2,3 +2,4 @@
 gradio==3.25.0
 numpy<2.0
 tensorflow-cpu==2.11.*
+keras==3.0.0