Spaces:
Running
Running
Add gallery
Browse files
app.py
CHANGED
@@ -17,6 +17,13 @@ def _fetch_url(*args, **kwargs):
|
|
17 |
print(f"Request error: {e}")
|
18 |
return None
|
19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
def generate(
|
21 |
prompt,
|
22 |
seed=None,
|
@@ -83,11 +90,25 @@ with gr.Blocks(
|
|
83 |
value=1024
|
84 |
)
|
85 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
86 |
btn.click(
|
87 |
generate,
|
88 |
inputs=[prompt, seed, randomize_seed, width, height],
|
89 |
outputs=[result],
|
90 |
api_name="run"
|
|
|
|
|
|
|
|
|
|
|
|
|
91 |
)
|
92 |
|
93 |
if __name__ == "__main__":
|
|
|
17 |
print(f"Request error: {e}")
|
18 |
return None
|
19 |
|
20 |
+
def add_to_gallery(image, gallery=None):
|
21 |
+
if gallery is None:
|
22 |
+
gallery = []
|
23 |
+
if image is not None:
|
24 |
+
gallery = gallery + [image]
|
25 |
+
return gallery
|
26 |
+
|
27 |
def generate(
|
28 |
prompt,
|
29 |
seed=None,
|
|
|
90 |
value=1024
|
91 |
)
|
92 |
|
93 |
+
with gr.Tab("🖼️ Gallery"):
|
94 |
+
gallery = gr.Gallery(
|
95 |
+
label="Gallery",
|
96 |
+
show_label=False,
|
97 |
+
format="jpeg",
|
98 |
+
interactive=False
|
99 |
+
)
|
100 |
+
|
101 |
btn.click(
|
102 |
generate,
|
103 |
inputs=[prompt, seed, randomize_seed, width, height],
|
104 |
outputs=[result],
|
105 |
api_name="run"
|
106 |
+
).then(
|
107 |
+
add_to_gallery,
|
108 |
+
inputs=[result, gallery],
|
109 |
+
outputs=[gallery],
|
110 |
+
queue=False,
|
111 |
+
show_label=False
|
112 |
)
|
113 |
|
114 |
if __name__ == "__main__":
|