kkk0123k commited on
Commit
321cd1e
·
verified ·
1 Parent(s): a665af8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,9 +14,9 @@ def classify_image(img):
14
  pred, idx, probs = learn.predict(img)
15
  return dict(zip(categories, map(float, probs)))
16
 
17
- image = gr.Image(shape=(192, 192))
18
  label = gr.Label()
19
  examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
20
 
21
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
22
- intf.launch(inline=False)
 
14
  pred, idx, probs = learn.predict(img)
15
  return dict(zip(categories, map(float, probs)))
16
 
17
+ image = gr.Image(type="pil", image_mode="RGB", size=(192, 192))
18
  label = gr.Label()
19
  examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
20
 
21
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
22
+ intf.launch(inline=False)