cyberai-1 commited on
Commit ·
dfe8843
1
Parent(s): cc8203a
Add
Browse files
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]
|
| 188 |
"probabilities": {
|
| 189 |
-
c: float(p
|
| 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 |
|