wooyeolbaek
commited on
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)
|