RamAnanth1 commited on
Commit
4d41f30
1 Parent(s): 7fc9487

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -12
app.py CHANGED
@@ -7,11 +7,11 @@ pipe = SemanticEditPipeline.from_pretrained(
7
  "runwayml/stable-diffusion-v1-5",
8
  ).to(device)
9
 
10
- def infer(prompt):
11
 
12
  gen = torch.Generator(device=device)
13
 
14
- gen.manual_seed(21)
15
  out = pipe(prompt=prompt, generator=gen, num_images_per_prompt=1, guidance_scale=7)
16
  images = out.images[0]
17
  out_edit = pipe(prompt=prompt, generator=gen, num_images_per_prompt=1, guidance_scale=7,
@@ -159,6 +159,7 @@ block = gr.Blocks(css=css)
159
  examples = [
160
  [
161
  'A photo of the face of a firefighter',
 
162
  ]
163
 
164
  ]
@@ -251,6 +252,7 @@ with block:
251
  label="Generated images", show_label=False, elem_id="gallery"
252
  ).style(height="auto")
253
 
 
254
  # with gr.Group(elem_id="container-advanced-btns"):
255
  # #advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
256
  # with gr.Group(elem_id="share-btn-container"):
@@ -258,20 +260,20 @@ with block:
258
  # loading_icon = gr.HTML(loading_icon_html)
259
  # share_button = gr.Button("Share to community", elem_id="share-btn")
260
 
261
- # seed = gr.Slider(
262
- # label="Seed",
263
- # minimum=0,
264
- # maximum=2147483647,
265
- # step=1,
266
- # randomize=True,
267
- # )
268
 
269
- ex = gr.Examples(examples=examples, fn=infer, inputs=[text], outputs=[gallery], cache_examples=True)
270
  ex.dataset.headers = [""]
271
 
272
 
273
- text.submit(infer, inputs=[text], outputs=[gallery])
274
- btn.click(infer, inputs=[text], outputs=[gallery])
275
 
276
 
277
 
 
7
  "runwayml/stable-diffusion-v1-5",
8
  ).to(device)
9
 
10
+ def infer(prompt,seed):
11
 
12
  gen = torch.Generator(device=device)
13
 
14
+ gen.manual_seed(seed)
15
  out = pipe(prompt=prompt, generator=gen, num_images_per_prompt=1, guidance_scale=7)
16
  images = out.images[0]
17
  out_edit = pipe(prompt=prompt, generator=gen, num_images_per_prompt=1, guidance_scale=7,
 
159
  examples = [
160
  [
161
  'A photo of the face of a firefighter',
162
+ 21
163
  ]
164
 
165
  ]
 
252
  label="Generated images", show_label=False, elem_id="gallery"
253
  ).style(height="auto")
254
 
255
+ with gr.Accordion("Advanced settings", open=False):
256
  # with gr.Group(elem_id="container-advanced-btns"):
257
  # #advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
258
  # with gr.Group(elem_id="share-btn-container"):
 
260
  # loading_icon = gr.HTML(loading_icon_html)
261
  # share_button = gr.Button("Share to community", elem_id="share-btn")
262
 
263
+ seed = gr.Slider(
264
+ label="Seed",
265
+ minimum=0,
266
+ maximum=2147483647,
267
+ step=1,
268
+ randomize=True,
269
+ )
270
 
271
+ ex = gr.Examples(examples=examples, fn=infer, inputs=[text, seed], outputs=[gallery], cache_examples=True)
272
  ex.dataset.headers = [""]
273
 
274
 
275
+ text.submit(infer, inputs=[text,seed], outputs=[gallery])
276
+ btn.click(infer, inputs=[text,seed], outputs=[gallery])
277
 
278
 
279