romain130492 commited on
Commit
abe7e26
1 Parent(s): 7a67f0e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ categories = ('Dog', 'Cat')
13
 
14
  def classify_image(img):
15
  pred, idx, probs = learn.predict(img)
16
- return dic(zip(categories, map(float, probs)))
17
 
18
 
19
  image = gr.inputs.Image(shape=(192, 192))
 
13
 
14
  def classify_image(img):
15
  pred, idx, probs = learn.predict(img)
16
+ return dict(zip(categories, map(float, probs)))
17
 
18
 
19
  image = gr.inputs.Image(shape=(192, 192))