Debug

handler.py

@@ -25,7 +25,7 @@ class EndpointHandler:
         device: The device (CPU or CUDA) used to run model inference.
     """
 
-    def __init__(self):
+    def __init__(self, path: str = ""):
         """
         Initializes the EndpointHandler with the model and processor from the current directory.
         """
@@ -40,10 +40,10 @@ class EndpointHandler:
         # Load the model and processor from the current directory
         self.logger.info("Loading model and processor from the current directory.")
         try:
-            self.processor = AutoImageProcessor.from_pretrained(os.getcwd())
-            self.model = AutoModel.from_pretrained(
-                os.getcwd(), trust_remote_code=True
-            ).to(self.device)
+            self.processor = AutoImageProcessor.from_pretrained(path)
+            self.model = AutoModel.from_pretrained(path, trust_remote_code=True).to(
+                self.device
+            )
             self.logger.info("Model and processor loaded successfully.")
         except Exception as e:
             self.logger.error(f"Failed to load model or processor: {e}")