Update app.py

app.py

@@ -26,12 +26,13 @@ else:
 @dataclass
 class Config:
     name: str = "flux-dev"
-    device: str = "cuda" if torch.cuda.is_available() else "cpu"
+    device: str = "cpu"
     offload: bool = False
     share: bool = False
     ckpt_dir: str = "."
     
 xflux_pipeline = XFluxPipeline(Config.name, Config.device, Config.offload)
+xflux_pipeline.to(device='cuda' if torch.cuda.is_available() else 'cpu')
 
 @spaces.GPU
 def generate(**kwargs):