Ignaciobfp commited on
Commit
a7e0386
·
1 Parent(s): aa31153

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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