Updated handler

handler.py

@@ -46,11 +46,10 @@ class EndpointHandler:
         # Register the custom class
         setattr(torch.nn.modules, 'BertSdpaSelfAttention', BertSdpaSelfAttention)
         
-        # Use the current directory if path is empty
-        if not path:
-            path = os.getcwd()
+        # Get the directory of the current script
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        model_path = os.path.join(current_dir, "model.joblib")
         
-        model_path = os.path.join(path, "model.joblib")
         self.model = self.load_model(model_path)
         self.tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")