akhaliq HF staff commited on
Commit
df760f0
·
1 Parent(s): 00c1359

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def inference(img):
62
  result = reversed(results[0].argsort()[-5:])
63
  resultdic = {}
64
  for r in result:
65
- resultdic[labels[str(r)]] = int(results[0][r])
66
  return resultdic
67
 
68
  gr.Interface(inference,gr.inputs.Image(type="filepath"),"label").launch()
 
62
  result = reversed(results[0].argsort()[-5:])
63
  resultdic = {}
64
  for r in result:
65
+ resultdic[labels[str(r)]] = float(results[0][r])
66
  return resultdic
67
 
68
  gr.Interface(inference,gr.inputs.Image(type="filepath"),"label").launch()