allispaul commited on
Commit
9486004
1 Parent(s): 278b1ca
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -47,17 +47,17 @@ def predict(img: Image) -> Tuple[dict, float]:
47
  title = "FoodVision Mini"
48
  description = "EfficientNetB2 feature extractor to classify images of food as pizza, steak, or sushi."
49
  article = "From the [Zero to Mastery PyTorch tutorial](https://www.learnpytorch.io/09_pytorch_model_deployment/)"
50
- examples = [list(example) for example in Path("examples").glob("*.jpg")]
51
 
52
  demo = gr.Interface(
53
  fn=predict,
54
  inputs=gr.Image(type="pil"),
55
  outputs=[gr.Label(num_top_classes=3, label="Predictions"),
56
  gr.Number(label="Prediction time (s)")],
57
- examples=example_list,
58
  title=title,
59
  description=description,
60
  article=article,
61
  )
62
 
63
- demo.lauch()
 
47
  title = "FoodVision Mini"
48
  description = "EfficientNetB2 feature extractor to classify images of food as pizza, steak, or sushi."
49
  article = "From the [Zero to Mastery PyTorch tutorial](https://www.learnpytorch.io/09_pytorch_model_deployment/)"
50
+ examples = [[example] for example in Path("examples").glob("*.jpg")]
51
 
52
  demo = gr.Interface(
53
  fn=predict,
54
  inputs=gr.Image(type="pil"),
55
  outputs=[gr.Label(num_top_classes=3, label="Predictions"),
56
  gr.Number(label="Prediction time (s)")],
57
+ examples=examples,
58
  title=title,
59
  description=description,
60
  article=article,
61
  )
62
 
63
+ demo.launch()