gokillmon commited on
Commit
a32f550
·
verified ·
1 Parent(s): b0c961c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ learn = load_learner('export.pkl')
8
  categories = ('gluten', 'lactose')
9
  def classify_image(img):
10
  pred, idx, probs = learn.predict(img)
11
- return dict(zip(categories, map(float, probs))) resources.get(pred)
12
 
13
  labels = ['gluten', 'lactose']
14
  urls = ['https://www.hopkinsmedicine.org/health/conditions-and-diseases/celiac-disease/what-is-a-glutenfree-diet', 'https://rebel-kitchen.com/blogs/news/what-does-lactose-free-actually-mean']
 
8
  categories = ('gluten', 'lactose')
9
  def classify_image(img):
10
  pred, idx, probs = learn.predict(img)
11
+ return dict(zip(categories, map(float, probs))), resources.get(pred)
12
 
13
  labels = ['gluten', 'lactose']
14
  urls = ['https://www.hopkinsmedicine.org/health/conditions-and-diseases/celiac-disease/what-is-a-glutenfree-diet', 'https://rebel-kitchen.com/blogs/news/what-does-lactose-free-actually-mean']