mart9992
/

vierundvi

Inference Endpoints

Model card Files Files and versions Community

mart9992 commited on Jan 5

Commit

f3c66d6

•

1 Parent(s): a8fb158

m

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -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)
         if mask_pil.mode != 'RGB':
             mask_pil = mask_pil.convert('RGB')

         just_get_sd_mask_function = just_get_sd_mask
     def __call__(self, data):
+        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')