Update app.py

app.py

@@ -6,10 +6,10 @@ import subprocess
 from pathlib import Path
 
 from dataclasses import dataclass
-import torch.multiprocessing as mp
 import spaces
 import gradio as gr
 import torch
+torch.multiprocessing.set_start_method('spawn')
 
 from src.flux.xflux_pipeline import XFluxPipeline
 
@@ -227,7 +227,6 @@ def parse_args() -> Config:
     return Config(**vars(args))
 
 if __name__ == "__main__":
-    mp.set_start_method("spawn")
     config = Config()
     demo = create_demo(config.name, config.device, config.offload, config.ckpt_dir)
     demo.launch(share=config.share)