Spaces:
Running
Running
smileycutie0
commited on
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__":
|