Update handler.py

handler.py

@@ -7,9 +7,8 @@ import os
 class InferenceHandler:
     def __init__(self):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        model_name = "colt12/maxcushion"
+        model_name = "./"  # Use the current directory
 
-        # Load the pipeline with authentication
         self.pipe = StableDiffusionXLPipeline.from_pretrained(
             model_name,
             torch_dtype=torch.float16,