Spaces:

yuragoithf
/

mlg_semantic_segmentation

Sleeping

yuragoithf commited on Jun 27, 2023

Commit

d3497cf

1 Parent(s): e32e754

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ def dice_coef(y_true, y_pred, smooth=1):
 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")
-image_output = gr.outputs.Image(type="pil", label="Output Image")
 # outputs = gr.outputs.HTML() #uncomment for single class output

 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")
+image_output = gr.outputs.Image(type="numpy", label="Output Image")
 # outputs = gr.outputs.HTML() #uncomment for single class output