Update app.py

app.py

@@ -67,7 +67,7 @@ def gen_pred(img, model=seg_model):
     img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img = img/255
-    img = tf.expand_dims(img, axis=0)
+    img = keras.ops.expand_dims(img, axis=0)
     pred = model.predict(img)
     pred = np.squeeze(pred, axis=0)
     fig = plt.figure(figsize=(3, 3))