Update app_demo.py

app_demo.py

@@ -14,7 +14,7 @@ from safetensors.torch import load_file
 from concurrent.futures import ThreadPoolExecutor
 import uuid
 #import cv2
-
+model_id = "Lykon/dreamshaper-7" #"openskyml/lcm-lora-sdxl-turbo" #"SimianLuo/LCM_Dreamshaper_v7"
 DESCRIPTION = '''# Fast Stable Diffusion CPU with Latent Consistency Model
 Distilled from [Dreamshaper v7](https://huggingface.co/Lykon/dreamshaper-7) fine-tune of [Stable Diffusion v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5) with only 4,000 training iterations (~32 A100 GPU Hours). [Project page](https://latent-consistency-models.github.io)
 '''
@@ -32,7 +32,7 @@ DTYPE = torch.float32  # torch.float16 works as well, but pictures seem to be a
 
 #"SimianLuo/LCM_Dreamshaper_v7"
 '''
-pipe = DiffusionPipeline.from_pretrained(  "openskyml/lcm-lora-sdxl-turbo"  ,    custom_pipeline="openskyml/lcm-lora-sdxl-turbo",
+pipe = DiffusionPipeline.from_pretrained(  model_id  ,    custom_pipeline=model_id,
     custom_revision="main",
     low_cpu_mem_usage=True,
     safety_checker= None,
@@ -45,7 +45,7 @@ pipe.to("cpu")
 
 #from diffusers import StableDiffusionPipeline
 
-model_id = "openskyml/lcm-lora-sdxl-turbo" #"SimianLuo/LCM_Dreamshaper_v7"
+
 
 pipe = StableDiffusionPipeline.from_pretrained(model_id)
 prompt = "A futuristic cityscape at sunset"