Dricz commited on
Commit
9e2718e
·
verified ·
1 Parent(s): 3fe3c0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -10,11 +10,8 @@ def response(image):
10
  # Plot results image
11
  im_bgr = r.plot() # BGR-order numpy array
12
  im_rgb = im_bgr[..., ::-1] # Convert BGR to RGB
13
- plt.figure(figsize=(10, 10))
14
- plt.imshow(im_rgb)
15
- plt.axis('off')
16
- plt.show()
17
- return results
18
 
19
  iface = gr.Interface(fn=response, inputs="image", outputs="image")
20
  iface.launch()
 
10
  # Plot results image
11
  im_bgr = r.plot() # BGR-order numpy array
12
  im_rgb = im_bgr[..., ::-1] # Convert BGR to RGB
13
+
14
+ return im_rgb
 
 
 
15
 
16
  iface = gr.Interface(fn=response, inputs="image", outputs="image")
17
  iface.launch()