Spaces:
Running
Running
Update app.py
Browse files
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():
|