download

app.py

@@ -49,6 +49,8 @@ def FRFsegment(input_img):
     
     mask = np.argmax(np.squeeze(est_label, axis=0),-1)
     pred = resize(mask, (w, h), preserve_range=True, clip=True)
+
+    imsave("label.png", pred)
     
     #overlay plot
     plt.clf()
@@ -56,11 +58,11 @@ def FRFsegment(input_img):
     plt.imshow(pred, alpha=0.4)
     plt.axis("off")
     plt.margins(x=0, y=0)
-    return plt
+    return plt, "label.png"
 
 
 title = "Segment beach imagery taken from a tower in Duck, NC, USA"
 description = "This model segments beach imagery into 4 classes: vegetation, sand, coarse sand, and background (water + sky + buildings + people)"
 examples = [['examples/FRF_c1_snap_20191112160000.jpg'], ['examples/FRF_c1_snap_20170101.jpg']] 
 
-FRFSegapp = gr.Interface(FRFsegment, gr.inputs.Image(), "image", examples=examples, title = title, description = description).launch()
+FRFSegapp = gr.Interface(FRFsegment, gr.inputs.Image(), ["image",gr.outputs.File()], examples=examples, title = title, description = description).launch()