Update app.py

app.py

@@ -149,8 +149,11 @@ def use_output_as_input(output_image):
     return gr.update()
 
 # Initialize Qwen Image Edit pipeline
-# Scheduler configuration for Lightning
+# --- Model Loading ---
 dtype = torch.bfloat16
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# Scheduler configuration for Lightning
 scheduler_config = {
     "base_image_seq_len": 256,
     "base_shift": math.log(3),