yuragoithf commited on
Commit
8371111
·
1 Parent(s): eafad1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="numpy", label="Upload an image", source="upload")
56
- inputs = gradio.inputs.ImageUpload()
57
  image_output = gr.outputs.Image(type="numpy", label="Output Image")
58
  # outputs = gr.outputs.HTML() #uncomment for single class output
59
 
 
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="numpy", label="Upload an image", source="upload")
56
+ inputs = gr.inputs.ImageUpload()
57
  image_output = gr.outputs.Image(type="numpy", label="Output Image")
58
  # outputs = gr.outputs.HTML() #uncomment for single class output
59