Clarice Groeneveld commited on
Commit
fe7bfcb
1 Parent(s): 19c37c9
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ def classify_image(img):
8
  pred,idx,probs = learn.predict(img)
9
  return(dict(zip(categories, map(float,probs))))
10
 
11
- image = gr.inputs.Image(shape=(192,192))
12
- label = gr.outputs.Label()
13
  examples = ['berry.jpg', 'chocolate.jpg', 'mint_choc.jpg', 'vanilla_soft.jpg']
14
 
15
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
 
8
  pred,idx,probs = learn.predict(img)
9
  return(dict(zip(categories, map(float,probs))))
10
 
11
+ image = gr.inputs.Image(shape=(192,192))
12
+ label = gr.components.Label()
13
  examples = ['berry.jpg', 'chocolate.jpg', 'mint_choc.jpg', 'vanilla_soft.jpg']
14
 
15
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)