Spaces:
Sleeping
Sleeping
Commit
·
a7e0386
1
Parent(s):
aa31153
Update app.py
Browse files
app.py
CHANGED
@@ -32,9 +32,10 @@ def predict(img):
|
|
32 |
mask = np.array(outputs.cpu())
|
33 |
mask[mask==1]=255
|
34 |
mask=np.reshape(mask,(400,400))
|
35 |
-
return Image.fromarray(mask.astype('uint8')
|
36 |
|
37 |
# Creamos la interfaz y la lanzamos.
|
38 |
#examples=['examples/1CA SUR_1200_800.png', 'examples/1CA SUR_4000_1200.png', 'examples/1CA SUR_4800_2000.png']
|
39 |
gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(400, 400)), outputs=gr.outputs.Image(type="pil")).launch(share=False)
|
|
|
40 |
|
|
|
32 |
mask = np.array(outputs.cpu())
|
33 |
mask[mask==1]=255
|
34 |
mask=np.reshape(mask,(400,400))
|
35 |
+
return Image.fromarray(mask.astype('uint8'))
|
36 |
|
37 |
# Creamos la interfaz y la lanzamos.
|
38 |
#examples=['examples/1CA SUR_1200_800.png', 'examples/1CA SUR_4000_1200.png', 'examples/1CA SUR_4800_2000.png']
|
39 |
gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(400, 400)), outputs=gr.outputs.Image(type="pil")).launch(share=False)
|
40 |
+
|
41 |
|