update pipeline

pipeline.py

@@ -2,13 +2,10 @@ from dataclasses import dataclass
 
 import cv2
 import numpy as np
-import rembg
 import torch
 import torch.nn.functional as F
 from diffusers import DiffusionPipeline
 from diffusers.utils import BaseOutput
-from PIL import Image
-from torchvision.transforms import v2
 
 
 def pad_camera_extrinsics_4x4(extrinsics):
@@ -99,15 +96,9 @@ class InstantMeshPipeline(DiffusionPipeline):
         self.lrm = lrm
         self.register_modules(lrm=self.lrm)
 
-        self.multi_view_diffusion = DiffusionPipeline.from_pretrained(
-            "dylanebert/zero123plus-instantmesh",
-            custom_pipeline="sudo-ai/zero123plus-pipeline",
-            torch_dtype=torch.float16,
-        )
-
     @torch.no_grad()
-    def __call__(self, image: Image.Image, remove_bg: bool = True):
-        if remove_bg:
+    def __call__(self, images: torch.Tensor):
+        """if remove_bg:
             image = rembg.remove(image)
 
         image = np.array(image)
@@ -156,7 +147,7 @@ class InstantMeshPipeline(DiffusionPipeline):
         images = images.unsqueeze(0)
         images = v2.functional.resize(
             images, 320, interpolation=3, antialias=True
-        ).clamp(0, 1)
+        ).clamp(0, 1)"""
 
         self.lrm.init_flexicubes_geometry(self._execution_device, fovy=30.0)
         cameras = get_zero123plus_input_cameras().to(self._execution_device)