Update handler.py

handler.py

@@ -11,13 +11,15 @@ logger = logging.getLogger(__name__)
 class EndpointHandler:
     def __init__(self, path=""):
         self.path = path
+        logger.info(f"Initializing handler with path: {path}")
+        logger.info(f"Directory contents: {os.listdir(path) if os.path.exists(path) else 'Path does not exist'}")
         try:
             self.task = self._determine_task()
         except Exception as e:
             logger.error(f"Failed to determine task: {str(e)}")
             raise
         
-        logger.info(f"Initializing model for task: {self.task} at path: {path}")
+        logger.info(f"Initializing model for task: {self.task}")
         if self.task == "text-generation":
             self.model = AutoModelForCausalLM.from_pretrained(
                 path, 
@@ -49,6 +51,7 @@ class EndpointHandler:
 
     def _determine_task(self):
         config_path = os.path.join(self.path, "config.json")
+        logger.info(f"Checking for config.json at: {config_path}")
         if not os.path.exists(config_path):
             logger.error(f"config.json not found in {self.path}")
             raise ValueError(f"config.json not found in {self.path}")