Update app.py

app.py

@@ -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=os.path.join(sys.path[0], "flagged")).launch(share=True)
+iface = gr.Interface(fn=segment, inputs=gr.inputs.Image(shape=(512, 512)), outputs="image", flagging_dir="flagged")).launch(share=True)