Commit ·
a6279fd
1
Parent(s): 5f4f504
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,10 @@ pipe = pipe.to(device)
|
|
| 21 |
|
| 22 |
|
| 23 |
def inference(prompt):
|
| 24 |
-
image = pipe(
|
|
|
|
|
|
|
|
|
|
| 25 |
net_attn_maps = get_net_attn_map(image.size)
|
| 26 |
net_attn_maps = resize_net_attn_map(net_attn_maps, image.size)
|
| 27 |
net_attn_maps = return_net_attn_map(net_attn_maps, pipe.tokenizer, prompt)
|
|
|
|
| 21 |
|
| 22 |
|
| 23 |
def inference(prompt):
|
| 24 |
+
image = pipe(
|
| 25 |
+
prompt,
|
| 26 |
+
num_inference_steps=15,
|
| 27 |
+
).images[0]
|
| 28 |
net_attn_maps = get_net_attn_map(image.size)
|
| 29 |
net_attn_maps = resize_net_attn_map(net_attn_maps, image.size)
|
| 30 |
net_attn_maps = return_net_attn_map(net_attn_maps, pipe.tokenizer, prompt)
|