Update handler.py

handler.py

@@ -7,11 +7,15 @@ class InferenceHandler:
     def __init__(self):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         model_name = "colt12/maxcushion"
+
+        # If your model is private, include the use_auth_token parameter
         self.pipe = StableDiffusionXLPipeline.from_pretrained(
             model_name,
             torch_dtype=torch.float16,
-            use_safetensors=True
-        ).to(self.device)  # Moved to the device here
+            use_safetensors=True,
+            # Uncomment the line below and replace with your token if needed
+            # use_auth_token="your_huggingface_token"
+        ).to(self.device)
 
     def __call__(self, inputs):
         prompt = inputs.get("prompt", "")