Fix

Image3DProcessor.py

@@ -11,6 +11,7 @@ from PIL import Image
 import rembg
 from omegaconf import OmegaConf
 from huggingface_hub import hf_hub_download
+from io import BytesIO
 from utils.app_utils import (
     remove_background,
     resize_foreground,
@@ -41,6 +42,10 @@ class Image3DProcessor:
         self.model.eval()
         
     def preprocess(self, input_image, preprocess_background=True, foreground_ratio=0.65):
+        # Convert bytes to PIL Image if necessary
+        if isinstance(input_image, bytes):
+            input_image = Image.open(BytesIO(input_image))
+        
         rembg_session = rembg.new_session()
         if preprocess_background:
             image = input_image.convert("RGB")
@@ -93,4 +98,3 @@ class Image3DProcessor:
             loop_renders.append(torch.clamp(rendered_image * 255, 0.0, 255.0).detach().permute(1, 2, 0).cpu().numpy().astype(np.uint8))
 
         return mesh_path, video_path
-