mrdbourke commited on
Commit
a0b862b
1 Parent(s): b3e1078

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  import torch
3
 
4
  from model import create_effnetb2_model
@@ -47,7 +48,7 @@ demo = gr.Interface(
47
  gr.Label(num_top_classes=3, label="Predictions"),
48
  gr.Number(label="Prediction time (s)"),
49
  ],
50
- # examples="demo/foodvision_mini/examples",
51
  interpretation="default",
52
  title=title,
53
  description=description,
 
1
  import gradio as gr
2
+ import os
3
  import torch
4
 
5
  from model import create_effnetb2_model
 
48
  gr.Label(num_top_classes=3, label="Predictions"),
49
  gr.Number(label="Prediction time (s)"),
50
  ],
51
+ examples=[["examples/" + example] for example in os.listdir("examples")],
52
  interpretation="default",
53
  title=title,
54
  description=description,