Laurits commited on
Commit
ecf73c9
·
1 Parent(s): 983810b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def classify_image(img):
8
  results = dict(zip(list(learn.dls.vocab),probs))
9
  best_results = sorted(results, key=results.get, reverse=True)[:5]
10
  probabilities = {key: float(results[key]) for key in best_results}
11
- return {probabilities}
12
 
13
  image = gr.inputs.Image(shape=(224,224))
14
  label = gr.outputs.Label()
 
8
  results = dict(zip(list(learn.dls.vocab),probs))
9
  best_results = sorted(results, key=results.get, reverse=True)[:5]
10
  probabilities = {key: float(results[key]) for key in best_results}
11
+ return probabilities
12
 
13
  image = gr.inputs.Image(shape=(224,224))
14
  label = gr.outputs.Label()