Spaces:

shivambhosale
/

spacenet3-unet-1024-1024

Runtime error

shivambhosale commited on Jul 6, 2022

Commit

769b12b

•

1 Parent(s): 44c54df

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import gradio as gr
 def getoutput(input_img):
     unet = UNet().to('cpu')
     unet = torch.load("unet_06_07_2022_17_13_42_256_256.pth", map_location='cpu').to('cpu')
-    output_img = make_predictions(unet, input_img, threshold=0.5)
     return output_img
 demo = gr.Interface(getoutput, gr.Image(shape=(200, 200)), "image")

 def getoutput(input_img):
     unet = UNet().to('cpu')
     unet = torch.load("unet_06_07_2022_17_13_42_256_256.pth", map_location='cpu').to('cpu')
+    output_img = make_predictions(unet, input_img, threshold=0.05)
     return output_img
 demo = gr.Interface(getoutput, gr.Image(shape=(200, 200)), "image")