ZhengPeng7 commited on
Commit
726f866
·
1 Parent(s): 92fbbd5

Add example images.

Browse files
app.py CHANGED
@@ -63,10 +63,15 @@ def pred_maps(dr):
63
  # cv2.imwrite('a.png', cv2.cvtColor(image_pred, cv2.COLOR_RGB2BGR))
64
  return image_preds[:]
65
 
 
 
 
 
 
66
  demo = gr.Interface(
67
  fn=pred_maps,
68
- inputs='text',
69
- outputs=['image', 'image', 'image', 'image', 'image'],
70
- css=".output_image, .input_image {height: 300px !important}",
71
  )
72
  demo.launch(debug=True)
 
63
  # cv2.imwrite('a.png', cv2.cvtColor(image_pred, cv2.COLOR_RGB2BGR))
64
  return image_preds[:]
65
 
66
+
67
+ examples = glob('example_images/butterfly')
68
+
69
+ ipt = ['image' for _ in range(5)]
70
+ opt = ipt.copy()
71
  demo = gr.Interface(
72
  fn=pred_maps,
73
+ inputs=ipt,
74
+ outputs=opt,
75
+ examples=examples,
76
  )
77
  demo.launch(debug=True)
example_images/butterfly/17057.jpg ADDED
example_images/butterfly/1757511.jpg ADDED
example_images/butterfly/18140.jpg ADDED
example_images/butterfly/18355.jpg ADDED
example_images/butterfly/22398.jpg ADDED
example_images/butterfly/22987.jpg ADDED
example_images/butterfly/22998.jpg ADDED
example_images/butterfly/354528.jpg ADDED
example_images/butterfly/3751159.jpg ADDED
example_images/butterfly/4013494.jpg ADDED
example_images/butterfly/4277400.jpg ADDED
example_images/butterfly/4320159.jpg ADDED
example_images/butterfly/4362807.jpg ADDED
example_images/butterfly/4375789.jpg ADDED
example_images/butterfly/4392735.jpg ADDED
example_images/butterfly/4672380.jpg ADDED