Spaces:
Runtime error
Runtime error
Commit
·
5f3c0c5
1
Parent(s):
f3ff7e0
updates files
Browse files
app.py
CHANGED
|
@@ -22,14 +22,10 @@ def infer(img, threshold):
|
|
| 22 |
pred = np.array(pred.numpy())[0,:,:,0]
|
| 23 |
return pred
|
| 24 |
|
| 25 |
-
gr_input = [
|
| 26 |
-
gr.inputs.Image(label="Image", type="numpy", shape=(512, 512))
|
| 27 |
-
,gr.inputs.Slider(minimum=0, maximum=1, step=0.05, default=0.5, label="Segmentation Threshold")
|
| 28 |
]
|
| 29 |
|
| 30 |
-
gr_output = [
|
| 31 |
-
gr.outputs.Image(type="pil",label="Segmentation Mask"),
|
| 32 |
-
# gr.outputs.Image(type="pil",label="Filtered Image"),
|
| 33 |
]
|
| 34 |
|
| 35 |
iface = gr.Interface(fn=infer, title='ribs segmentation model', description='Keras implementation of ResUNET++ for xray ribs segmentation', inputs=gr_input, outputs=gr_output, examples=examples, flagging_dir="flagged").launch(cache_examples=True)
|
|
|
|
| 22 |
pred = np.array(pred.numpy())[0,:,:,0]
|
| 23 |
return pred
|
| 24 |
|
| 25 |
+
gr_input = [gr.inputs.Image(label="Image", type="numpy", shape=(512, 512)), gr.inputs.Slider(minimum=0, maximum=1, step=0.05, default=0.5, label="Segmentation Threshold")
|
|
|
|
|
|
|
| 26 |
]
|
| 27 |
|
| 28 |
+
gr_output = [gr.outputs.Image(type="pil",label="Segmentation Mask"),
|
|
|
|
|
|
|
| 29 |
]
|
| 30 |
|
| 31 |
iface = gr.Interface(fn=infer, title='ribs segmentation model', description='Keras implementation of ResUNET++ for xray ribs segmentation', inputs=gr_input, outputs=gr_output, examples=examples, flagging_dir="flagged").launch(cache_examples=True)
|