Update handler.py

handler.py

@@ -30,9 +30,13 @@ class EndpointHandler:
         Args:
             path (str): Path to the GGUF file.
         """
-        files_and_dirs = os.listdir(".")  # "." represents the current directory
-        print(files_and_dirs)
-        self.model = Llama(model_path="./Llama-3.2-1B-Instruct-Q4_K_L.gguf")
+        # Absolute or relative path to the model file
+        model_path = path if path else "./Llama-3.2-1B-Instruct-Q4_K_L.gguf"
+        
+        if not os.path.exists(model_path):
+            raise ValueError(f"Model path does not exist: {model_path}")
+
+        self.model = Llama(model_path=model_path)
         self.tokenizer = self.model.tokenizer  # GGUF-specific tokenizer, if available
 
     def __call__(self, data: Any) -> List[Dict[str, str]]: