Update handler.py

handler.py

@@ -11,8 +11,8 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 if device.type != 'cuda':
     raise ValueError("Need to run on GPU")
 
-class SDXLInpaintHandler:
-    def __init__(self, path="mrcuddle/urpm-inpaint-sdxl"):
+class EndpointHandler:
+    def __init__(self, path="mrcuddle/URPM-Inpaint-Hyper-SDXL"):
         """Load the SDXL Inpainting model."""
         self.pipeline = StableDiffusionXLInpaintPipeline.from_pretrained(
             path, torch_dtype=torch.float16
@@ -69,7 +69,8 @@ class SDXLInpaintHandler:
         image.save(buffered, format="PNG")
         return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
-handler = SDXLInpaintHandler()
+# Create an instance of EndpointHandler
+handler = EndpointHandler()
 
 def handle(data: dict):
     return handler(data)