Update app.py

app.py

@@ -65,7 +65,7 @@ def generate_image(prompt, ckpt):
         images, has_nsfw_concepts = check_nsfw_images(results.images)
         if any(has_nsfw_concepts):
             gr.Warning("NSFW content detected.")
-            return Image.new("RGB", (512, 512))
+            return Image.new("RGB", (512, 256))
         return images[0]
     return results.images[0]
 
@@ -84,7 +84,7 @@ with gr.Blocks(css="style.css") as demo:
     img = gr.Image(label='SDXL-Lightning Generated Image')
 
     prompt.submit(fn=generate_image,
-                 inputs=[prompt, ckpt],
+                 inputs=["header for blog post about: "+prompt, ckpt],
                  outputs=img,
                  )
     submit.click(fn=generate_image,