Update app.py

app.py

@@ -27,6 +27,7 @@ import shutil
 import uuid
 import zipfile
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
 # META: CUDA_CHECK / GPU_INFO
 print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
 print("torch.__version__ =", torch.__version__)
@@ -80,7 +81,6 @@ loras = [
 
 # Initialize the base model
 dtype = torch.bfloat16
-device = "cuda" if torch.cuda.is_available() else "cpu"
 base_model = "Qwen/Qwen-Image"
 
 # Scheduler configuration from the Qwen-Image-Lightning repository