feat: add class name in predict v2

main.py

@@ -109,13 +109,6 @@ def plot_detected_rectangles(image, detections, output_path):
     cv2.imwrite(output_path, img_with_rectangles)
 
 
-# # Call the animal_detect_and_classify function to get detections
-# detections = animal_detect_and_classify('/content/cat_tiger.jpg')
-
-# # Plot the detected rectangles with their corresponding class names
-# plot_detected_rectangles(cv2.imread('/content/cat_tiger.jpg'), detections)
-
-
 @app.post("/predict/v2")
 async def predict_v2(file: UploadFile = File(...)):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_")
@@ -124,10 +117,12 @@ async def predict_v2(file: UploadFile = File(...)):
     image = Image.open(BytesIO(contents))
     image.save("input/" + filename)
     detections = animal_detect_and_classify("input/" + filename)
+    class_names = [class_name for _, class_name in detections]
     plot_detected_rectangles(cv2.imread("input/" + filename), detections, "output/" + filename)
     return {
         "message": "Detection and classification completed successfully",
-        "data": "output/" + filename
+        "data": "output/" + filename,
+        "class_names": class_names
     }
 
 @app.get("/image/")