Update app.py

app.py

@@ -22,7 +22,7 @@ import tempfile
 from functools import partial
 from optimum.quanto import quantize, qfloat8, freeze
 from diffusers import FluxPipeline
-
+import torch.multiprocessing as mp
 from src.utils.train_util import instantiate_from_config
 from src.utils.camera_util import (
     FOV_to_intrinsics, 
@@ -31,6 +31,7 @@ from src.utils.camera_util import (
 )
 from src.utils.mesh_util import save_obj, save_glb
 from src.utils.infer_util import remove_background, resize_foreground, images_to_video
+mp.set_start_method('spawn')
 
 # Set up cache path
 cache_path = path.join(path.dirname(path.abspath(__file__)), "models")