Update app.py

app.py

@@ -180,7 +180,9 @@ if IS_SPACE:
     downloader = CheckpointDownloader()
     downloader.download_all_checkpoints()
 
-pipeline = load_pipeline(use_distilled=False, device="cuda")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline = load_pipeline(use_distilled=False, device=device)
+
 class HunyuanImageApp:
 
     @space_context(duration=290)