Update handler.py

handler.py

@@ -11,7 +11,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 if device.type != 'cuda':
     raise ValueError("Need to run on GPU")
 
-class SDXLInpaintHandler:
+class EndpointHandler:
     def __init__(self, path="mrcuddle/URPM-Inpaint-Hyper-SDXL"):
         """Load the SDXL Inpainting model."""
         self.pipeline = StableDiffusionXLInpaintPipeline.from_pretrained(
@@ -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)