cutiee82 commited on
Commit
ddbca23
·
verified ·
1 Parent(s): 3ace83d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def _fetch(*args, retries=5, **kwargs):
27
 
28
  def _load_examples():
29
  ds = load_dataset("k-mktr/improved-flux-prompts", split="train")
30
- return [item["prompt"] for item in ds[:20]]
31
 
32
  def generate(
33
  prompt,
 
27
 
28
  def _load_examples():
29
  ds = load_dataset("k-mktr/improved-flux-prompts", split="train")
30
+ return ds.shuffle()["prompt"][:20]
31
 
32
  def generate(
33
  prompt,