MakeNTU-v2024
/

Image-preprocessor

Model card Files Files and versions

oscarfu0501 commited on May 4, 2024

Commit

53c4179

·

verified ·

1 Parent(s): 70698c4

Update handler.py

Files changed (1) hide show

handler.py +3 -1

handler.py CHANGED Viewed

@@ -4,10 +4,12 @@ import numpy as np
 import cv2
 import base64
 from ultralytics import YOLO
 class EndpointHandler:
     def __init__(self, path='.'):  # pass api key to model
         self.model = YOLO("../best.pt")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:

 import cv2
 import base64
 from ultralytics import YOLO
+import os
 class EndpointHandler:
     def __init__(self, path='.'):  # pass api key to model
+        current_directory = os.getcwd()
+        print("Current working directory:", current_directory)
         self.model = YOLO("../best.pt")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: