Spaces:
Running
on
Zero
Running
on
Zero
Shaamallow
commited on
Commit
•
2e581a8
1
Parent(s):
183c72e
update demo with examples
Browse files
app.py
CHANGED
@@ -96,7 +96,6 @@ def invert_image(model, image: np.ndarray, n_steps: int, width: int, height: int
|
|
96 |
image_tensor = image_tensor.unsqueeze(0)
|
97 |
latent = img_to_latents(image_tensor, model.vae)
|
98 |
|
99 |
-
print(latent)
|
100 |
model.set_ip_adapter_scale(0)
|
101 |
|
102 |
inv_latents = model(
|
@@ -175,11 +174,19 @@ def generate(
|
|
175 |
return image
|
176 |
|
177 |
|
178 |
-
|
179 |
-
"Astronaut in a jungle,
|
180 |
-
"
|
|
|
|
|
|
|
|
|
181 |
]
|
182 |
|
|
|
|
|
|
|
|
|
183 |
with gr.Blocks(css=css) as demo:
|
184 |
gr.Markdown(DESCRIPTION)
|
185 |
|
@@ -299,7 +306,7 @@ with gr.Blocks(css=css) as demo:
|
|
299 |
|
300 |
gr.Examples(
|
301 |
examples=examples,
|
302 |
-
inputs=prompt,
|
303 |
outputs=result,
|
304 |
fn=generate,
|
305 |
)
|
|
|
96 |
image_tensor = image_tensor.unsqueeze(0)
|
97 |
latent = img_to_latents(image_tensor, model.vae)
|
98 |
|
|
|
99 |
model.set_ip_adapter_scale(0)
|
100 |
|
101 |
inv_latents = model(
|
|
|
174 |
return image
|
175 |
|
176 |
|
177 |
+
examples_prompts = [
|
178 |
+
"Astronaut in a jungle, detailed, 8k",
|
179 |
+
"A Bird",
|
180 |
+
"A Tiger",
|
181 |
+
"A Cat",
|
182 |
+
"cactus",
|
183 |
+
"A Panda",
|
184 |
]
|
185 |
|
186 |
+
examples_images = [f"./images/{i}.png" for i in range(6)]
|
187 |
+
|
188 |
+
examples = [[prompt, image] for prompt, image in zip(examples_prompts, examples_images)]
|
189 |
+
|
190 |
with gr.Blocks(css=css) as demo:
|
191 |
gr.Markdown(DESCRIPTION)
|
192 |
|
|
|
306 |
|
307 |
gr.Examples(
|
308 |
examples=examples,
|
309 |
+
inputs=[prompt, style_image],
|
310 |
outputs=result,
|
311 |
fn=generate,
|
312 |
)
|