Update handler.py

handler.py

@@ -5,6 +5,7 @@ import base64
 import torch
 import csv
 from transformers import CLIPProcessor, CLIPModel
+import os 
 
 
 class EndpointHandler():
@@ -14,7 +15,7 @@ class EndpointHandler():
         self.model.eval()
 
         # Load categories from CSV
-        self.categories = self.load_categories_from_csv("categories.csv")
+        self.categories = self.load_categories_from_csv(os.path.join(path, "categories.csv"))
 
     def load_categories_from_csv(self, filepath: str) -> List[str]:
         categories = []