Spaces:
Sleeping
Sleeping
Fix
Browse files
app.py
CHANGED
@@ -82,7 +82,7 @@ def predict(image: PIL.Image.Image, score_threshold: float,
|
|
82 |
probs = model.predict(image[None, ...])[0]
|
83 |
probs = probs.astype(float)
|
84 |
res = dict()
|
85 |
-
for prob, label in zip(probs, labels):
|
86 |
if prob < score_threshold:
|
87 |
continue
|
88 |
res[label] = prob
|
|
|
82 |
probs = model.predict(image[None, ...])[0]
|
83 |
probs = probs.astype(float)
|
84 |
res = dict()
|
85 |
+
for prob, label in zip(probs.tolist(), labels):
|
86 |
if prob < score_threshold:
|
87 |
continue
|
88 |
res[label] = prob
|