ludjan commited on
Commit
d655f9b
1 Parent(s): e601273

Add failing example

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -17,13 +17,8 @@ def classify_image(img):
17
  is_tori_logo, idx, probs = learn.predict(img)
18
  return dict(zip(categories, map(float,probs)))
19
 
20
- examples = [
21
- 'images/dog.jpeg',
22
- 'images/cat.jpeg',
23
- 'images/face.jpg',
24
- 'images/profile-avatar.jpeg',
25
- 'images/tori-logo.png'
26
- ]
27
 
28
  image = gr.Image()
29
  label = gr.Label()
 
17
  is_tori_logo, idx, probs = learn.predict(img)
18
  return dict(zip(categories, map(float,probs)))
19
 
20
+ image_dir = 'images'
21
+ examples = map(lambda image_name: f"{image_dir}/{image_name}", os.listdir(image_dir))
 
 
 
 
 
22
 
23
  image = gr.Image()
24
  label = gr.Label()