Update app.py

app.py

@@ -9,7 +9,7 @@ def load_model():
         "sd-community/sdxl-flash",
         torch_dtype=torch.float16,
         low_cpu_mem_usage=True
-    )#.to("cuda")
+    ).to(0)
     pipe.scheduler = DPMSolverSinglestepScheduler.from_config(
         pipe.scheduler.config,
         timestep_spacing="trailing"