RebeccaNissan26 commited on
Commit
6abe857
·
1 Parent(s): ed0545d

small change to examples line

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,7 +82,7 @@ if __name__ == "__main__":
82
  section_btn.click(partial(predict, model=model), img_input, img_output)
83
 
84
  images_dir = glob(os.path.join(os.getcwd(), "images") + os.sep + "*.png")
85
- examples = [i for i in np.random.choice(images_dir, size=1, replace=False)]
86
  gr.Examples(examples=examples, inputs=img_input, outputs=img_output)
87
 
88
  demo.launch()
 
82
  section_btn.click(partial(predict, model=model), img_input, img_output)
83
 
84
  images_dir = glob(os.path.join(os.getcwd(), "images") + os.sep + "*.png")
85
+ examples = [i for i in images_dir]
86
  gr.Examples(examples=examples, inputs=img_input, outputs=img_output)
87
 
88
  demo.launch()