Update app.py

app.py

@@ -37,6 +37,7 @@ def inference_mask1(
         mask = torch.from_numpy(np.stack(prompt[i]['layers'], axis=0).any(0).any(-1)).cpu()
         mask[mask > 0] = 1
 
+        return (mask.detach().cpu().numpy() * 255).astype(np.uint8)
         if mask.sum() == 0:
             break
         mask = F.interpolate(mask.unsqueeze(0).unsqueeze(0).float(), query_img.size()[-2:], mode='nearest').squeeze(0).squeeze(0)