Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ learn = load_learner('fast-food-model.pkl')
|
|
8 |
|
9 |
categories = ('Baked Potato', 'Burger', 'Crispy Chicken', 'Donut', 'Fries','Hot Dog','Pizza','Sandwich','Taco','Taquito')
|
10 |
def classify_image (image):
|
11 |
-
pred,idx,
|
12 |
return dict(zip(categories, map(float, probs)))
|
13 |
|
14 |
input_image = gradio.inputs.Image(shape=(192, 192))
|
|
|
8 |
|
9 |
categories = ('Baked Potato', 'Burger', 'Crispy Chicken', 'Donut', 'Fries','Hot Dog','Pizza','Sandwich','Taco','Taquito')
|
10 |
def classify_image (image):
|
11 |
+
pred,idx,probs = learn.predict(image)
|
12 |
return dict(zip(categories, map(float, probs)))
|
13 |
|
14 |
input_image = gradio.inputs.Image(shape=(192, 192))
|