Update app.py

app.py

@@ -51,14 +51,15 @@ pipe_edit.to("cuda")
 
 # Image Generator
 # Keep the models loaded globally for reuse
-pipe = StableDiffusionXLPipeline.from_pretrained(
+if torch.cuda.is_available():
+    pipe = StableDiffusionXLPipeline.from_pretrained(
         "fluently/Fluently-XL-v4",
         torch_dtype=torch.float16,
         use_safetensors=True,
     ).to("cuda")
-pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
-pipe.set_adapters("dalle")
+    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
+    pipe.set_adapters("dalle")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed: