Spaces:

amoghv
/

Fast-food-classifier

Sleeping

amoghv commited on Dec 24, 2022

Commit

3390643

1 Parent(s): 02d8da5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ learn = load_learner('fast-food-model.pkl')
 categories = ('Baked Potato', 'Burger', 'Crispy Chicken', 'Donut', 'Fries','Hot Dog','Pizza','Sandwich','Taco','Taquito')
 def classify_image (image):
-    pred,idx,prob = learn.predict(image)
     return dict(zip(categories, map(float, probs)))
 input_image = gradio.inputs.Image(shape=(192, 192))

 categories = ('Baked Potato', 'Burger', 'Crispy Chicken', 'Donut', 'Fries','Hot Dog','Pizza','Sandwich','Taco','Taquito')
 def classify_image (image):
+    pred,idx,probs = learn.predict(image)
     return dict(zip(categories, map(float, probs)))
 input_image = gradio.inputs.Image(shape=(192, 192))