Spaces:
Runtime error
Runtime error
Commit
·
1fa2651
1
Parent(s):
b503ff5
Update app.py
Browse files
app.py
CHANGED
@@ -21,4 +21,4 @@ def segment(img_arr):
|
|
21 |
pred = tf.cast(prob > threshold, dtype=tf.float32)
|
22 |
pred = np.array(pred.numpy())[0,:,:,0]
|
23 |
return pred
|
24 |
-
iface = gr.Interface(fn=segment, inputs=gr.inputs.Image(shape=(512, 512)), outputs="image"
|
|
|
21 |
pred = tf.cast(prob > threshold, dtype=tf.float32)
|
22 |
pred = np.array(pred.numpy())[0,:,:,0]
|
23 |
return pred
|
24 |
+
iface = gr.Interface(fn=segment, inputs=gr.inputs.Image(shape=(512, 512)), outputs="image").launch()
|