Update app.py

app.py

@@ -104,10 +104,10 @@ print("Loading Model!")
 schedule: str = "euler_a"
 base_model_path: str = "stabilityai/stable-diffusion-xl-base-1.0"
 refiner_model_path: str = "stabilityai/stable-diffusion-xl-refiner-1.0" 
-unet_path: str = "cosmicman/CosmicMan-SDXL"
+# unet_path: str = "cosmicman/CosmicMan-SDXL"
 SCHEDULER = schedule_map[schedule]
 scheduler = SCHEDULER.from_pretrained(base_model_path, subfolder="scheduler", torch_dtype=torch.float16)
-# unet = UNet2DConditionModel.from_pretrained(unet_path, torch_dtype=torch.float16)
+unet = UNet2DConditionModel.from_pretrained(unet_path, torch_dtype=torch.float16)
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
     base_model_path,
@@ -193,7 +193,7 @@ with gr.Blocks(theme=gr.themes.Soft(),css="style.css") as demo:
         inputs=input_prompt,
         outputs=result,
         fn=generate_image,
-        cache_examples=CACHE_EXAMPLES,
+        cache_examples=0,
     )
 
     gr.on(