charlesjavelona commited on
Commit
d5b5f16
·
1 Parent(s): d1523b9
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,10 +9,10 @@ learn = load_learner('model.pkl')
9
 
10
  def classify_image(img):
11
  pred, idx, probs = learn.predict(img)
12
- return dict(zip(categories, map(float, probs))
13
 
14
 
15
- image = gr.inputs.Image(192, 192)
16
  label = gr.outputs.Label()
17
  examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
18
 
 
9
 
10
  def classify_image(img):
11
  pred, idx, probs = learn.predict(img)
12
+ return dict(zip(categories, map(float, probs)))
13
 
14
 
15
+ image = gr.inputs.Image(shape=(192, 192))
16
  label = gr.outputs.Label()
17
  examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
18