Hugging Face's logo

Spaces:
Sa-m
/
Brand-Logo-Classification
Running

App Files Community
Sa-m commited on
Commit
db99f1d
·
1 Parent(s): 786704b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def classify_image(inp):
40
 
41
  label_order = ["Burger King", "KFC", "McDonalds", "Other", "Starbucks", "Subway"]
42
 
43
- result = {label: f"{prediction[0][labels[label]]:.2f}" for label in label_order}
44
 
45
  return result
46
 
 
40
 
41
  label_order = ["Burger King", "KFC", "McDonalds", "Other", "Starbucks", "Subway"]
42
 
43
+ result = {label: {prediction[0][labels[label]]:.2f} for label in label_order}
44
 
45
  return result
46