veb-101 commited on
Commit
4fe41eb
·
1 Parent(s): d50a6b4

updated num. examples

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ if __name__ == "__main__":
71
  section_btn.click(partial(predict, model=model, preprocess_fn=preprocess, device=DEVICE), img_input, img_output)
72
 
73
  images_dir = glob(os.path.join(os.getcwd(), "samples") + os.sep + "*.png")
74
- examples = [i for i in np.random.choice(images_dir, size=8, replace=False)]
75
  gr.Examples(examples=examples, inputs=img_input, outputs=img_output)
76
 
77
  demo.launch()
 
71
  section_btn.click(partial(predict, model=model, preprocess_fn=preprocess, device=DEVICE), img_input, img_output)
72
 
73
  images_dir = glob(os.path.join(os.getcwd(), "samples") + os.sep + "*.png")
74
+ examples = [i for i in np.random.choice(images_dir, size=10, replace=False)]
75
  gr.Examples(examples=examples, inputs=img_input, outputs=img_output)
76
 
77
  demo.launch()