Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import cv2
 import numpy as np
+import tensorflow as tf
 from tensorflow.keras.models import load_model
 from tensorflow.keras.applications.mobilenet_v2 import preprocess_input
 
 # Load model
-model = load_model("mask_mobilenet.h5")
+model = load_model("mask_mobilenet_fixed.h5", compile=False)
 
 IMG_SIZE = 224
 
@@ -15,7 +16,7 @@ def predict_mask(image):
     img = preprocess_input(img)
     img = np.expand_dims(img, axis=0)
 
-    prob = model.predict(img)[0][0]
+    prob = model.predict(img, verbose=0)[0][0]
 
     return {
         "Mask 😷": float(1 - prob),