Update app.py

app.py

@@ -1,3 +1,9 @@
+import spaces
+
+import mu
+import torch
+torch.multiprocessing.set_start_method('spawn', force=True)
+
 import re
 import os
 import yaml
@@ -6,11 +12,8 @@ import subprocess
 from pathlib import Path
 
 from dataclasses import dataclass
-import spaces
-import gradio as gr
-import torch
-torch.multiprocessing.set_start_method('spawn')
 
+import gradio as gr
 from src.flux.xflux_pipeline import XFluxPipeline
 
 import os