listen2you ysharma HF Staff commited on
Commit
31ed7e3
·
verified ·
1 Parent(s): 89a0b55

fixing examples (#4)

Browse files

- fixing examples (bd26aa390ecc504bc473f3c2656127f1ada8731c)


Co-authored-by: yuvraj sharma <[email protected]>

Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -33,8 +33,8 @@ examples = [
33
  ["examples 2/poster_orig.jpg", "replace 'lambs' with 'llamas'",("examples 2/poster_orig.jpg","examples 2/poster.jpg")],
34
  ]
35
 
36
- def generate_examples(init_image, output_image, prompt):
37
- return init_image, output_image, prompt
38
 
39
 
40
  def load_state_dict(model, ckpt_path, device="cuda", strict=False, assign=True):
@@ -483,10 +483,10 @@ with gr.Blocks() as demo:
483
 
484
  gr.Examples(
485
  examples,
486
- inputs=[init_image, prompt, output_image],
487
- outputs=[init_image, prompt, output_image],
488
  fn=generate_examples,
489
- cache_examples=False
490
  )
491
 
492
  demo.launch()
 
33
  ["examples 2/poster_orig.jpg", "replace 'lambs' with 'llamas'",("examples 2/poster_orig.jpg","examples 2/poster.jpg")],
34
  ]
35
 
36
+ def generate_examples(init_image, prompt):
37
+ return inference(prompt, init_image, seed=-1, size_level=512)
38
 
39
 
40
  def load_state_dict(model, ckpt_path, device="cuda", strict=False, assign=True):
 
483
 
484
  gr.Examples(
485
  examples,
486
+ inputs=[init_image, prompt],
487
+ outputs=[output_image, output_random_seed],
488
  fn=generate_examples,
489
+ cache_examples=True
490
  )
491
 
492
  demo.launch()