fix device

app.py

@@ -26,22 +26,20 @@ with open("sdxl_lora.json", "r") as file:
 # Sort the loras by likes
 sdxl_loras_raw = sorted(sdxl_loras_raw, key=lambda x: x["likes"], reverse=True)
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 model_id = "stabilityai/stable-diffusion-xl-base-1.0"
 
 if gr.NO_RELOAD:
-    torch.cuda.max_memory_allocated(device=device)
     pipe = DiffusionPipeline.from_pretrained(model_id, variant="fp16")
     pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
     pipe.load_lora_weights("jasperai/flash-sdxl", adapter_name="lora")
-    pipe.to(device="cuda", dtype=torch.float16)
+    pipe.to(device=DEVICE, dtype=torch.float16)
 
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 
-@spaces.GPU
 def check_and_load_lora_user(user_lora_selector, user_lora_weight, gr_lora_loaded):
     flash_sdxl_id = "jasperai/flash-sdxl"
 
@@ -277,7 +275,7 @@ with gr.Blocks(css=css) as demo:
             guidance_scale,
         ],
         outputs=[result],
-        show_progress=True,
+        show_progress="minimal",
     )
 
     user_lora_weight.change(
@@ -294,7 +292,7 @@ with gr.Blocks(css=css) as demo:
             user_lora_selector,
             pre_prompt,
         ],
-        show_progress=False,
+        show_progress="hidden",
     )
 
     gr.Markdown("**Disclaimer:**")