Update handler.py

handler.py

@@ -13,9 +13,9 @@ if device.type != 'cuda':
     raise ValueError("need to run on GPU")
 
 class EndpointHandler():
-    def __init__(self):
+    def __init__(self, path="viba98/materialic"):
         # load the optimized model
-        self.pipe = StableDiffusionPipeline.from_pretrained("viba98/materialic", torch_dtype=torch.float16)
+        self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
         self.pipe = self.pipe.to(device)