yuragoithf commited on
Commit
e5ecda2
·
1 Parent(s): 7f5ab8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -57,10 +57,11 @@ def predict_class(image):
57
  def classify_image(image):
58
  results = predict_class(image)
59
  output = {}
60
- for _, prediction in enumerate(results):
61
- predicted_label = labels.get(_)
62
- score = prediction[_]
63
  output[predicted_label] = score
 
64
  return output
65
 
66
 
 
57
  def classify_image(image):
58
  results = predict_class(image)
59
  output = {}
60
+ for index in range(len(results)):
61
+ predicted_label = labels.get(index)
62
+ score = results[index]
63
  output[predicted_label] = score
64
+ print(output)
65
  return output
66
 
67