Update app.py

app.py

@@ -13,8 +13,9 @@ from utils.planner import (
 # ----------------------------
 # 🖥️ Device Setup
 # ----------------------------
-device = "cuda" if torch.cuda.is_available() else "cpu"
-dtype = torch.float16 if device == "cuda" else torch.float32
+use_gpu = torch.cuda.is_available()
+device = "cuda" if use_gpu else "cpu"
+dtype = torch.float16 if use_gpu else torch.float32
 
 # ----------------------------
 # 🧠 Load SDXL Img2Img Pipeline (no ControlNet)
@@ -22,7 +23,7 @@ dtype = torch.float16 if device == "cuda" else torch.float32
 pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     torch_dtype=dtype,
-    variant="fp16" if device == "cuda" else None,
+    variant="fp16" if use_gpu else None,
     use_safetensors=True,
 )
 pipe.to(device)
@@ -94,3 +95,4 @@ demo = gr.Interface(
 if __name__ == "__main__":
     demo.launch()
 
+