cyberai-1 commited on
Commit
dfe8843
·
1 Parent(s): cc8203a
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -184,9 +184,9 @@ def predict():
184
  return jsonify({
185
  "class": CLASSES[pred_idx],
186
  #"confidence": float(probs[pred_idx]),
187
- "confidence": float(probs[pred_idx] * 100),
188
  "probabilities": {
189
- c: float(p * 100) for c, p in zip(CLASSES, probs)
190
  },
191
  })
192
 
 
184
  return jsonify({
185
  "class": CLASSES[pred_idx],
186
  #"confidence": float(probs[pred_idx]),
187
+ "confidence": float(probs[pred_idx]),
188
  "probabilities": {
189
+ c: float(p) for c, p in zip(CLASSES, probs)
190
  },
191
  })
192