yuragoithf commited on
Commit
07a61bc
·
1 Parent(s): 7b06bd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -52,7 +52,8 @@ def dice_coef(y_true, y_pred, smooth=1):
52
  # Load the model
53
  seg_model = tf.keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'dice_coef': dice_coef})
54
 
55
- inputs = gr.inputs.Image(type="pil", label="Upload an image", source="upload")
 
56
  image_output = gr.outputs.Image(type="numpy", label="Output Image")
57
  # outputs = gr.outputs.HTML() #uncomment for single class output
58
 
 
52
  # Load the model
53
  seg_model = tf.keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'dice_coef': dice_coef})
54
 
55
+ # inputs = gr.inputs.Image(type="pil", label="Upload an image", source="upload")
56
+ inputs = Image.open("./003e2c95d.jpg")
57
  image_output = gr.outputs.Image(type="numpy", label="Output Image")
58
  # outputs = gr.outputs.HTML() #uncomment for single class output
59