Update handler.py

handler.py

@@ -10,7 +10,13 @@ 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")
+        files_and_directories = os.listdir()
+
+        # Print each file and directory
+        for item in files_and_directories:
+            print(item)
+            
+        self.model = YOLO("./best.pt")
         
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         inputs = data.get("inputs")