Defer torch/bpy imports to GPU context

app.py

@@ -12,8 +12,6 @@ from pathlib import Path
 import gradio as gr
 import numpy as np
 import spaces
-import torch
-import bpy
 from huggingface_hub import snapshot_download, login
 
 # Login with HF_TOKEN if available
@@ -42,20 +40,21 @@ FACES = None
 def load_model():
     """Load SAM 3D Body model"""
     global MODEL, FACES
-    
+
     if MODEL is not None:
         return MODEL, FACES
-    
+
+    import torch
     print("Loading SAM 3D Body model...")
-    
+
     # Download checkpoint
     checkpoint_dir = snapshot_download(
         repo_id="facebook/sam-3d-body-dinov3",
         token=os.environ.get("HF_TOKEN")
     )
-    
+
     from sam_3d_body import load_sam_3d_body, SAM3DBodyEstimator
-    
+
     device = "cuda" if torch.cuda.is_available() else "cpu"
     
     model, model_cfg = load_sam_3d_body(
@@ -74,21 +73,23 @@ def load_model():
     return MODEL, FACES
 
 
-@spaces.GPU(duration=60)
+@spaces.GPU(duration=120)
 def reconstruct_body(image: np.ndarray) -> tuple:
     """
     Reconstruct 3D body mesh from image.
-    
+
     Args:
         image: Input RGB image
-    
+
     Returns:
         tuple: (glb_path, status)
     """
     if image is None:
         return None, "❌ No image provided"
-    
+
     try:
+        import torch
+        import bpy
         estimator, faces = load_model()
         
         # Process image