Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def inference(prompt):
|
|
30 |
num_inference_steps=15,
|
31 |
).images[0]
|
32 |
|
33 |
-
total_attn_maps = save_attention_maps(attn_maps, pipe.tokenizer,
|
34 |
|
35 |
return image, total_attn_maps
|
36 |
|
|
|
30 |
num_inference_steps=15,
|
31 |
).images[0]
|
32 |
|
33 |
+
total_attn_maps = save_attention_maps(attn_maps, pipe.tokenizer, prompt)
|
34 |
|
35 |
return image, total_attn_maps
|
36 |
|