Update app.py

app.py

@@ -26,11 +26,11 @@ pipe = pipe.to(device)
 
 def inference(prompt):
     image = pipe(
-        prompt,
+        [prompt],
         num_inference_steps=15,
     ).images[0]
 
-    total_attn_maps = save_attention_maps(attn_maps, pipe.tokenizer, prompt)
+    total_attn_maps = save_attention_maps(attn_maps, pipe.tokenizer, [prompt])
     
     return image, total_attn_maps