Update app.py

app.py

@@ -22,7 +22,7 @@ IMG_SCALING = (1, 1)
 #     return output
 
 #model_file = "./seg_unet_model.h5"
-model_file = "seg_unet_model.h5"
+model_file = 'seg_unet_model.h5'
 
 #Custom objects for model
 
@@ -51,7 +51,7 @@ def focal_loss_fixed(y_true, y_pred, gamma=2.0, alpha=0.25):
     return focal_loss_fixed
 
 # Load the model
-seg_model = keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'focal_loss_fixed': focal_loss_fixed, 'dice_coef': dice_coef})
+seg_model = keras.models.load_model(model_file, custom_objects={'Combo_loss': Combo_loss, 'focal_loss_fixed': focal_loss_fixed, 'dice_coef': dice_coef})
 
 # inputs = gr.inputs.Image(type="pil", label="Upload an image")
 # image_output = gr.outputs.Image(type="pil", label="Output Image")