Spaces:

mostafapasha
/

ribs-segmentation-space

Runtime error

mostafapasha commited on May 14, 2022

Commit

743224a

1 Parent(s): 08aebd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,4 +21,4 @@ def segment(img_arr):
     pred = tf.cast(prob > threshold, dtype=tf.float32)
     pred = np.array(pred.numpy())[0,:,:,0]
     return pred
-iface = gr.Interface(fn=segment, inputs=gr.inputs.Image(shape=(512, 512)), outputs="image", flagging_dir="flagged")).launch(share=True)

     pred = tf.cast(prob > threshold, dtype=tf.float32)
     pred = np.array(pred.numpy())[0,:,:,0]
     return pred
+iface = gr.Interface(fn=segment, inputs=gr.inputs.Image(shape=(512, 512)), outputs="image", flagging_dir="flagged").launch(share=True)