Update app.py

app.py

@@ -7,7 +7,7 @@ import numpy
 import uuid
 import json
 import os
-from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
+from diffusers import StableDiffusionXLPipeline, ControlNetModel, EulerAncestralDiscreteScheduler
 from PIL import Image
 
 # Pre-Initialize
@@ -26,7 +26,7 @@ DEFAULT_WIDTH = 1024
 REPO = "sd-community/sdxl-flash"
 
 model = StableDiffusionXLPipeline.from_pretrained(REPO, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
-model.scheduler = EulerAncestralDiscreteScheduler.from_config(model.scheduler.config)
+#model.scheduler = EulerAncestralDiscreteScheduler.from_config(model.scheduler.config)
 model.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
 model.set_adapters(["base"], adapter_weights=[0.7])
 model.to(DEVICE)