vkthakur88 commited on
Commit
d9e4675
1 Parent(s): a814421

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ def predict(image, label_list):
12
  labels
13
  )
14
 
15
- return response[0]
 
16
 
17
  app = gr.Interface(
18
  fn = predict,
 
12
  labels
13
  )
14
 
15
+ result = {v['label'] : v['score'] for v in response}
16
+ return result
17
 
18
  app = gr.Interface(
19
  fn = predict,