Update app_demo.py

app_demo.py

@@ -30,17 +30,25 @@ DTYPE = torch.float32  # torch.float16 works as well, but pictures seem to be a
 #pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
 
 
-
-pipe = DiffusionPipeline.from_pretrained(    "SimianLuo/LCM_Dreamshaper_v7",    custom_pipeline="latent_consistency_txt2img",
+#"SimianLuo/LCM_Dreamshaper_v7"
+pipe = DiffusionPipeline.from_pretrained(  "charliebaby2023/p_y"  ,    custom_pipeline="latent_consistency_txt2img",
     custom_revision="main",
     low_cpu_mem_usage=True
-    pipe.safety_checker = None
+    pipe.safety_checker = None,
+ torch_dtype="float16", use_safetensors=True
 )
 pipe.to(torch_device="cpu", torch_dtype=DTYPE)
 pipe.to("cpu")
 
 
+from diffusers import StableDiffusionPipeline
+
+model_id = "your-huggingface-username/your-safetensors-model"
 
+#pipe = StableDiffusionPipeline.from_pretrained(model_id,)
+#prompt = "A futuristic cityscape at sunset"
+#image = pipeline(prompt).images[0]
+#image.show()
 
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: