Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -63,7 +63,7 @@ def classify_image(model, image_processor, class_info, device, image_url, accura
|
|
| 63 |
class_name = class_info['class_columns'][i]
|
| 64 |
predicted_classes.append(f"{class_name}: {prob:.3f}")
|
| 65 |
predicted_list.append(class_name)
|
| 66 |
-
prob_list.append(prob)
|
| 67 |
|
| 68 |
# Mostrar resultado
|
| 69 |
if predicted_classes:
|
|
@@ -77,7 +77,7 @@ def classify_image(model, image_processor, class_info, device, image_url, accura
|
|
| 77 |
class_name = class_info['class_columns'][max_idx]
|
| 78 |
print(f"{class_name}: {max_prob:.3f}")
|
| 79 |
predicted_list.append(class_name)
|
| 80 |
-
prob_list.append(max_prob)
|
| 81 |
return {"class": predicted_list, "accuracy":prob_list}
|
| 82 |
|
| 83 |
class EndpointHandler():
|
|
|
|
| 63 |
class_name = class_info['class_columns'][i]
|
| 64 |
predicted_classes.append(f"{class_name}: {prob:.3f}")
|
| 65 |
predicted_list.append(class_name)
|
| 66 |
+
prob_list.append(float(prob))
|
| 67 |
|
| 68 |
# Mostrar resultado
|
| 69 |
if predicted_classes:
|
|
|
|
| 77 |
class_name = class_info['class_columns'][max_idx]
|
| 78 |
print(f"{class_name}: {max_prob:.3f}")
|
| 79 |
predicted_list.append(class_name)
|
| 80 |
+
prob_list.append(float(max_prob))
|
| 81 |
return {"class": predicted_list, "accuracy":prob_list}
|
| 82 |
|
| 83 |
class EndpointHandler():
|