updated num. examples
Browse files
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=
|
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()
|