add required packages

__init__.py

@@ -1,6 +1,26 @@
 from transformers import AutoProcessor
 from .modeling import LRMGenerator, LRMGeneratorConfig
-from .processor import LRMImageProcessor
 
-# we add the preprocessor
-AutoProcessor.register(LRMImageProcessor)
+import logging
+
+# we set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# helper function to install packages
+def install_package(package_name):
+    if importlib.util.find_spec(package_name) is None:
+        logger.info(f"Package '{package_name}' not found. Installing...")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
+    else:
+        logger.info(f"Package '{package_name}' is already installed.")
+
+# list of packages to ensure are installed
+required_packages = [
+    'imageio[ffmpeg]', 'PyMCubes', 'trimesh', 'rembg[gpu,cli]', 'kiui', 'torchvision', 'Pillow'
+]
+
+for package in required_packages:
+    install_package(package)
+
+from .processor import LRMImageProcessor