Update app.py

app.py

@@ -4,10 +4,22 @@ import tempfile
 import gradio as gr
 from PIL import Image
 from rembg import remove
+import sys
 import subprocess
 from glob import glob
 import requests
 
+# Ensure the required package is installed
+def install_dependencies():
+    try:
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d.git@75ebeeaea0908c5527e7b1e305fbc7681382db47"])
+    except subprocess.CalledProcessError as e:
+        print(f"Error installing dependencies: {e}")
+        sys.exit(1)  # Exit the script if installation fails
+
+# Install dependencies at the start
+install_dependencies()
+
 def remove_background(input_url):
     # Create a temporary folder for downloaded and processed images
     temp_dir = tempfile.mkdtemp()