slf188 commited on
Commit
f4c5e97
·
1 Parent(s): 962b07b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ categories = ('Cat', 'Dog')
10
 
11
  def classify_image(img):
12
  pred, idx, probs = learn.predict(img)
13
- return dict(zip(categories), map(float, probs))
14
 
15
  image = gr.inputs.Image(shape=(192,192))
16
  label = gr.outputs.Label()
 
10
 
11
  def classify_image(img):
12
  pred, idx, probs = learn.predict(img)
13
+ return dict(zip(categories, map(float, probs)))
14
 
15
  image = gr.inputs.Image(shape=(192,192))
16
  label = gr.outputs.Label()