Update app.py

app.py

@@ -32,6 +32,7 @@ def end_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     shutil.rmtree(user_dir)
 
+@spaces.GPU
 def preprocess_image(image: Image.Image) -> Image.Image:
     """
     Preprocess the input image for 3D generation.
@@ -49,6 +50,7 @@ def preprocess_image(image: Image.Image) -> Image.Image:
     processed_image = pipeline.preprocess_image(image)
     return processed_image
 
+@spaces.GPU
 def preprocess_videos(video: str) -> List[Tuple[Image.Image, str]]:
     """
     Preprocess the input video for multi-image 3D generation.
@@ -76,6 +78,7 @@ def preprocess_videos(video: str) -> List[Tuple[Image.Image, str]]:
     processed_images = [pipeline.preprocess_image(image) for image in images]
     return processed_images
 
+@spaces.GPU
 def preprocess_images(images: List[Tuple[Image.Image, str]]) -> List[Image.Image]:
     """
     Preprocess a list of input images for multi-image 3D generation.
@@ -434,4 +437,5 @@ if __name__ == "__main__":
     pipeline = TrellisVGGTTo3DPipeline.from_pretrained("Stable-X/trellis-vggt-v0-1")
     pipeline.cuda()
     pipeline.VGGT_model.cuda()
+    pipeline.birefnet_model.cuda()
     demo.launch()