cutiee82 commited on
Commit
6c2c08f
·
verified ·
1 Parent(s): 0b465e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import time
3
  from io import BytesIO
4
  from urllib.parse import quote
5
 
 
6
  import gradio as gr
7
  import requests
8
  from PIL import Image
@@ -24,6 +25,10 @@ def _fetch(*args, retries=5, **kwargs):
24
  return None
25
  return None
26
 
 
 
 
 
27
  def generate(
28
  prompt,
29
  seed=None,
@@ -71,6 +76,8 @@ with gr.Blocks(
71
  container=False
72
  )
73
  btn = gr.Button("Generate", variant="primary")
 
 
74
 
75
  with gr.Tab("⚙️ Settings"):
76
  with gr.Row():
 
3
  from io import BytesIO
4
  from urllib.parse import quote
5
 
6
+ from datasets import load_dataset
7
  import gradio as gr
8
  import requests
9
  from PIL import Image
 
25
  return None
26
  return None
27
 
28
+ def _load_examples():
29
+ ds = load_dataset("k-mktr/improved-flux-prompts", split="train")
30
+ return [item["prompt"] for item in ds.shuffle()[:20]]
31
+
32
  def generate(
33
  prompt,
34
  seed=None,
 
76
  container=False
77
  )
78
  btn = gr.Button("Generate", variant="primary")
79
+
80
+ examples = gr.Examples(_load_examples(), inputs=[prompt])
81
 
82
  with gr.Tab("⚙️ Settings"):
83
  with gr.Row():