Update handler.py

handler.py

@@ -6,15 +6,15 @@ from transformers.image_utils import to_numpy_array, PILImageResampling, Channel
 from transformers.image_transforms import resize, to_channel_dimension_format
 
 class EndpointHandler:
-    def __init__(self, model_path: str, api_token: str):
+    def __init__(self, model_path: str):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.processor = AutoProcessor.from_pretrained(
             model_path,
-            token=api_token
+            # token=api_token
         )
         self.model = AutoModelForCausalLM.from_pretrained(
             model_path,
-            token=api_token,
+            # token=api_token,
             trust_remote_code=True,
             torch_dtype=torch.bfloat16,
         ).to(self.device)