Update app.py

app.py

@@ -1,6 +1,5 @@
 import spaces
 import torch
-# torch.multiprocessing.set_start_method('spawn', force=True)
 
 import re
 import os
@@ -36,6 +35,7 @@ xflux_pipeline.to(device='cuda' if torch.cuda.is_available() else 'cpu')
 
 @spaces.GPU
 def generate(**kwargs): 
+    torch.multiprocessing.set_start_method('spawn', force=True)
     return xflux_pipeline.gradio_generate(**kwargs)
     
 def parse_args() -> Config: