pemujo commited on
Commit
cb6539f
·
1 Parent(s): d708432

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ if __name__ == "__main__":
89
  section_btn = gr.Button("Generate Predictions")
90
  section_btn.click(partial(predict, model=model), img_input, img_output)
91
 
92
- images_dir = glob(os.path.join(os.getcwd(), "images") + os.sep + "*.png")
93
  examples = [i for i in images_dir]
94
  gr.Examples(examples=examples, inputs=img_input, outputs=img_output)
95
 
 
89
  section_btn = gr.Button("Generate Predictions")
90
  section_btn.click(partial(predict, model=model), img_input, img_output)
91
 
92
+ images_dir = glob(os.path.join(os.getcwd(), "images") + os.sep + "*.jpg")
93
  examples = [i for i in images_dir]
94
  gr.Examples(examples=examples, inputs=img_input, outputs=img_output)
95