Update app.py

app.py

@@ -7,7 +7,7 @@ import gradio as gr
 import numpy as np
 import random
 #from diffusers import DiffusionPipeline
-from diffusers import StableDiffusionXLPipeline, DDIMScheduler
+from diffusers import StableDiffusionXLPipeline, TCDScheduler
 from huggingface_hub import hf_hub_download
 
 
@@ -34,9 +34,13 @@ pipe.to('cuda')
 pipe.load_lora_weights(hf_hub_download(repo_name, ckpt_name))
 pipe.fuse_lora()
 # Ensure ddim scheduler timestep spacing set as trailing !!!
-pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
+pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 # lower eta results in more detail
 
+pipe.load_lora_weights(repo_name)
+pipe.fuse_lora()
+
+
 prompt = "1girl, solo, upper body, v, smile, looking at viewer, outdoors, night, masterpiece, best quality, very aesthetic, absurdres"
 negative_prompt = "(worst quality),(low quality),lowres,(bad anatomy),(deformed anatomy),(deformed fingers),(blurry),(extra finger),(extra arms), (extra legs),(monochrome:1.4),(grayscale:1.4),((watermark)),(overweight female:1.6),((pointy ears)),mascot,stuffed human, stuffed animal,chibi,english text, chinese text, korean text"