Update app.py

app.py

@@ -48,14 +48,12 @@ def dice_coef(y_true, y_pred, smooth=1):
 # Load the model
 seg_model = tf.keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'dice_coef': dice_coef})
 
-
-
-
-
 inputs = gr.inputs.Image(type="pil", label="Upload an image")
 # outputs = gr.outputs.HTML() #uncomment for single class output 
 
 def gen_pred(img=inputs, model=seg_model):
+    if !img:
+        img = cv2.imread('003e2c95d.jpg')
     #rgb_path = os.path.join(test_image_dir,img)
     #img = cv2.imread(rgb_path)
     img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]