Update processor.py

processor.py

@@ -3,35 +3,24 @@ import importlib
 import sys
 import logging
 from transformers import AutoProcessor
-print("lib added")
-
-
-# # a helper function to install packages
-# def install_package(package_name):
-#     if importlib.util.find_spec(package_name) is None:
-#         print(f"Package '{package_name}' not found. Installing...")
-#         subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
-#     else:
-#         print(f"Package '{package_name}' is already installed.")
-# # we ensure the necessary packages are installed
-# packages = [
-#     'imageio[ffmpeg]', 'PyMCubes', 'trimesh', 'rembg[gpu,cli]', 'kiui', 'torchvision', 'Pillow'
-# ]
-# for package in packages:
-#     install_package(package)
-
-# only then we import the packages after installation
-print("importing")
 import torch
 import numpy as np
 from rembg import remove, new_session
 from functools import partial
-print("import torch")
 from torchvision.utils import save_image
 from PIL import Image
 from kiui.op import recenter
 import kiui
 
+print("lib added")
+
+# commented out the package installation part as it's not necessary for this fix
+# and might be causing issues if run repeatedly
+
+print("importing")
+
+print("import torch")
+
 class LRMImageProcessor(AutoProcessor):
     def __init__(self, source_size=512, *args, **kwargs):
         print("add super")
@@ -68,7 +57,10 @@ class LRMImageProcessor(AutoProcessor):
         fx, fy, cx, cy = self.get_normalized_camera_intrinsics(intrinsics)
         return torch.cat([
             RT.reshape(-1, 12),
-            fx.unsqueeze(-1), fy.unsqueeze(-1), cx.unsqueeze(-1), cy.unsqueeze(-1),
+            fx.unsqueeze(-1),
+            fy.unsqueeze(-1),
+            cx.unsqueeze(-1),
+            cy.unsqueeze(-1),
         ], dim=-1)
 
     def _default_intrinsics(self):
@@ -100,4 +92,4 @@ class LRMImageProcessor(AutoProcessor):
         processed_image = self.preprocess_image(image)
         print("processed done")
         source_camera = self._default_source_camera(batch_size=1)
-        return processed_image, source_camera
+        return processed_image, source_camera