img_pil issue

app.py

@@ -59,6 +59,7 @@ def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image, k:
         k = 8
 
     # Prepare tensor
+    img_pil = base_image.convert("RGB")
     img_resized = img_pil.resize((384, 384))
     img_tensor = torch.from_numpy(np.array(img_resized)).permute(2, 0, 1).unsqueeze(0).float() / 255.0
     mean = torch.as_tensor([0.5, 0.5, 0.5], device=img_tensor.device).view(1, 3, 1, 1)