Update app.py

app.py

@@ -8,7 +8,7 @@ from diffusers import DiffusionPipeline
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3.1-aesthetic", torch_dtype=dtype).to(device)
+pipe = DiffusionPipeline.from_pretrained("aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -108,7 +108,7 @@ Shuttle 3.1 Aesthetic is a text-to-image AI model designed to create aesthetic,
             fn = infer,
             inputs = [prompt],
             outputs = [result, seed],
-            cache_examples="lazy"
+            cache_examples=False
         )
 
     gr.on(