alubart commited on
Commit
37c4ab7
1 Parent(s): 198d0f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -15,13 +15,14 @@ learn = load_learner('model.pkl')
15
  categories = ('Dog', 'Cat')
16
 
17
  def classify_image(img):
18
- pred,idx,probs = learn.predict(img)
 
 
 
19
  return dict(zip(categories, map(float,probs)))
20
 
21
  # Cell
22
- image = gr.inputs.Image(shape=(192, 192))
23
- label = gr.outputs.Label()
24
  examples = ['dog.jpeg', 'cat.jpeg', 'elephant.jpeg']
25
 
26
- intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
27
  intf.launch(inline=False)
 
15
  categories = ('Dog', 'Cat')
16
 
17
  def classify_image(img):
18
+ im = PILImage.create(img)
19
+ im.thumbnail((192,192))
20
+
21
+ pred,idx,probs = learn.predict(im)
22
  return dict(zip(categories, map(float,probs)))
23
 
24
  # Cell
 
 
25
  examples = ['dog.jpeg', 'cat.jpeg', 'elephant.jpeg']
26
 
27
+ intf = gr.Interface(fn=classify_image, inputs="image", outputs="label", examples=examples)
28
  intf.launch(inline=False)