Commit
·
2b32190
1
Parent(s):
79ff307
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def dice_coef(y_true, y_pred, smooth=1):
|
|
| 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(label="Output Image")
|
| 57 |
# outputs = gr.outputs.HTML() #uncomment for single class output
|
| 58 |
|
| 59 |
def gen_pred(img=inputs, model=seg_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 |
|
| 59 |
def gen_pred(img=inputs, model=seg_model):
|