Update app.py

Addes Dalle lora for better image (test)

app.py

@@ -8,7 +8,7 @@ import numpy as np
 from PIL import Image
 import spaces
 import torch
-from diffusers import DiffusionPipeline
+from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
@@ -22,21 +22,17 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 if torch.cuda.is_available():
-    pipe = DiffusionPipeline.from_pretrained(
+    pipe = StableDiffusionXLPipeline.from_pretrained(
         "sd-community/sdxl-flash",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False
     )
-    if ENABLE_CPU_OFFLOAD:
-        pipe.enable_model_cpu_offload()
-    else:
-        pipe.to(device)       
-        print("Loaded on Device!")
-    
-    if USE_TORCH_COMPILE:
-        pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
-        print("Model Compiled!")
+    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.to("cuda")
 
 
 def save_image(img):