Timjo88 commited on
Commit
df84583
·
1 Parent(s): ec26715

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ labels = learn.dls.vocab
7
 
8
  def predict(img):
9
  pred,idx,probs = learn.predict(img)
10
- return {labels[i]: float(probs[i]) for i,_ in enumerate(labels))}
11
 
12
  image = gr.inputs.Image(shape=(256,256))
13
  label = gr.outputs.Label()
 
7
 
8
  def predict(img):
9
  pred,idx,probs = learn.predict(img)
10
+ return {labels[i]: float(probs[i]) for i,_ in enumerate(labels)}
11
 
12
  image = gr.inputs.Image(shape=(256,256))
13
  label = gr.outputs.Label()