Spaces:
Running
on
Zero
Running
on
Zero
fixing examples
#4
by
ysharma
HF Staff
- opened
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,
|
| 37 |
-
return init_image,
|
| 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
|
| 487 |
-
outputs=[
|
| 488 |
fn=generate_examples,
|
| 489 |
-
cache_examples=
|
| 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()
|