Update app.py

app.py

@@ -3,7 +3,7 @@ import tensorflow as tf
 import numpy as np
 import cv2
 
-# Load your SavedModel folder
+# Load the SavedModel
 model = tf.keras.models.load_model("mask_mobilenet_savedmodel")
 
 def predict_mask(image):
@@ -17,8 +17,9 @@ def predict_mask(image):
         image = image / 255.0
         image = np.expand_dims(image, axis=0)
 
-        # Predict using SavedModel callable
-        preds = model(image, training=False).numpy()
+        # Use the 'call' method instead of calling the object
+        preds = model.call(tf.convert_to_tensor(image), training=False).numpy()
+
         result = "Mask" if preds[0][0] > 0.5 else "No Mask"
         return result