vietngth commited on
Commit
5452d67
Β·
1 Parent(s): b38e70b

Update code to match new gradio

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ def classify_image(img):
16
  pred, idx, probs = loaded_learn.predict(img)
17
  return dict(zip(categories, map(float, probs)))
18
 
19
- image = gr.inputs.Image(shape=(192, 192))
20
- label = gr.outputs.Label()
21
  examples = ["doggo.jpg", "catto.jpg", "test.jpg"]
22
 
23
  intf = gr.Interface(
 
16
  pred, idx, probs = loaded_learn.predict(img)
17
  return dict(zip(categories, map(float, probs)))
18
 
19
+ image = gr.components.Image(shape=(192, 192))
20
+ label = gr.components.Label()
21
  examples = ["doggo.jpg", "catto.jpg", "test.jpg"]
22
 
23
  intf = gr.Interface(