Update app.py

app.py

@@ -27,7 +27,7 @@ def _fetch(*args, retries=5, **kwargs):
 
 def _load_examples():
     ds = load_dataset("k-mktr/improved-flux-prompts", split="train")
-    return ds.shuffle()["prompt"][:20]
+    return [[item] for item in ds.shuffle()["prompt"][:20]]
 
 def generate(
     prompt,