cutiee82 commited on
Commit
eb4340e
·
verified ·
1 Parent(s): a80fd23

Add gallery

Browse files
Files changed (1) hide show
  1. app.py +21 -0
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__":