Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
|
@@ -89,7 +89,7 @@ async def predict(file: UploadFile = File(...)):
|
|
| 89 |
preds = model.predict(img_array, verbose=0)
|
| 90 |
index = np.argmax(preds[0])
|
| 91 |
predicted_label = CLASS_NAMES[index]
|
| 92 |
-
confidence = float(np.max(preds[0]))
|
| 93 |
|
| 94 |
probability = {}
|
| 95 |
for i, prob in enumerate(preds[0]):
|
|
|
|
| 89 |
preds = model.predict(img_array, verbose=0)
|
| 90 |
index = np.argmax(preds[0])
|
| 91 |
predicted_label = CLASS_NAMES[index]
|
| 92 |
+
confidence = round(float(np.max(preds[0])) * 100, 2)
|
| 93 |
|
| 94 |
probability = {}
|
| 95 |
for i, prob in enumerate(preds[0]):
|