m

handler.py

@@ -49,7 +49,7 @@ class EndpointHandler():
         just_get_sd_mask_function = just_get_sd_mask
 
     def __call__(self, data):
-        mask_pil = just_get_sd_mask(Image.open("assets/demo1.jpg"), "bear", 10)
+        mask_pil = just_get_sd_mask_function(Image.open("assets/demo1.jpg"), "bear", 10)
 
         if mask_pil.mode != 'RGB':
             mask_pil = mask_pil.convert('RGB')