Commit
·
7a03b44
1
Parent(s):
c97d496
Update app.py
Browse files
app.py
CHANGED
|
@@ -48,8 +48,8 @@ def dice_coef(y_true, y_pred, smooth=1):
|
|
| 48 |
# Load the model
|
| 49 |
seg_model = tf.keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'dice_coef': dice_coef})
|
| 50 |
|
| 51 |
-
inputs = gr.inputs.Image(type="pil", label="Upload an image")
|
| 52 |
-
image_output = gr.Image()
|
| 53 |
# outputs = gr.outputs.HTML() #uncomment for single class output
|
| 54 |
|
| 55 |
def gen_pred(img=inputs, model=seg_model):
|
|
|
|
| 48 |
# Load the model
|
| 49 |
seg_model = tf.keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'dice_coef': dice_coef})
|
| 50 |
|
| 51 |
+
inputs = gr.inputs.Image(type="pil", label="Upload an image", source="upload")
|
| 52 |
+
image_output = gr.outputs.Image(type="pil", label="Output Image")
|
| 53 |
# outputs = gr.outputs.HTML() #uncomment for single class output
|
| 54 |
|
| 55 |
def gen_pred(img=inputs, model=seg_model):
|