Commit
·
07a61bc
1
Parent(s):
7b06bd6
Update app.py
Browse files
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 |
|