wooyeolbaek commited on
Commit
52edbc2
·
verified ·
1 Parent(s): 798dafc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, prompts)
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