Update app.py

app.py

@@ -34,18 +34,19 @@ def inference(prompt):
 
 with gr.Blocks() as demo:
     gr.Markdown(
-    """
-    🚀 Text-to-Image Cross Attention Map for 🧨 Diffusers ⚡
-    """)
+        """
+        # 🚀 Text-to-Image Cross Attention Map for 🧨 Diffusers ⚡
+        """
+    )
     prompt = gr.Textbox(value="A photo of a black puppy, christmas atmosphere", label="Prompt", lines=2)
     btn = gr.Button("Generate images", scale=0)
     
     with gr.Row():
         image = gr.Image(height=512,width=512,type="pil")
         gallery = gr.Gallery(
-        value=None,
-        label="Generated images", show_label=False, elem_id="gallery",
-        object_fit="contain", height="auto")
+            value=None, label="Generated images", show_label=False,
+            elem_id="gallery", object_fit="contain", height="auto"
+        )
     
 
     btn.click(inference, prompt, [image, gallery])