Update app.py

app.py

@@ -1619,7 +1619,7 @@ if __name__ == "__main__":
 
     # GRADIO MODE
 
-    def infer(prompt):
+    def infer(prompt, progress=gr.Progress()):
         set_seed(args.seed)
         width,height = list(map(int, args.resolution.split(',')))
         pipe = AccDiffusionSDXLPipeline.from_pretrained(args.model_ckpt, torch_dtype=torch.float16).to("cuda")
@@ -1672,8 +1672,8 @@ if __name__ == "__main__":
     }
     """
     with gr.Blocks(css=css) as demo:
-        with gr.Column():
-            gr.Markdown("# AccDiffusion")
+        with gr.Column(elem_id="col-container"):
+            gr.Markdown("# AccDiffusion: An Accurate Method for Higher-Resolution Image Generation")
             prompt = gr.Textbox(label="Prompt")
             submit_btn = gr.Button("Submit")
             output_images = gr.Gallery(format="png")