Update handler.py
Browse files- handler.py +1 -2
handler.py
CHANGED
|
@@ -75,8 +75,7 @@ def classify_image(model, image_processor, class_info, device, image_url, accura
|
|
| 75 |
class_name = class_info['class_columns'][max_idx]
|
| 76 |
print(f"{class_name}: {max_prob:.3f}")
|
| 77 |
predicted_list.append({"class": class_name, "confidence": float(max_prob)})
|
| 78 |
-
|
| 79 |
-
return predicted_list
|
| 80 |
|
| 81 |
class EndpointHandler():
|
| 82 |
def __init__(self, path=""):
|
|
|
|
| 75 |
class_name = class_info['class_columns'][max_idx]
|
| 76 |
print(f"{class_name}: {max_prob:.3f}")
|
| 77 |
predicted_list.append({"class": class_name, "confidence": float(max_prob)})
|
| 78 |
+
return predicted_list
|
|
|
|
| 79 |
|
| 80 |
class EndpointHandler():
|
| 81 |
def __init__(self, path=""):
|