Sa-m commited on
Commit
962146a
·
1 Parent(s): 1f1c04d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def classify_image(inp):
20
  prediction = model.predict(inp)
21
  label = dict((v,k) for k,v in labels.items())
22
  predicted_class_indices=np.argmax(prediction,axis=1)
23
- return {label[i]: float(predicted_class_indices[i]) for i in range(NUM_CLASSES)}
24
 
25
 
26
 
 
20
  prediction = model.predict(inp)
21
  label = dict((v,k) for k,v in labels.items())
22
  predicted_class_indices=np.argmax(prediction,axis=1)
23
+ return {labels[i]: float(predicted_class_indices[i]) for i in range(NUM_CLASSES)}
24
 
25
 
26