Update app.py

app.py

@@ -1663,7 +1663,7 @@ if __name__ == "__main__":
                     )
         print(images)
 
-        return "done"
+        return images
     
     css = """
     #col-container{
@@ -1675,9 +1675,8 @@ if __name__ == "__main__":
         with gr.Column():
             gr.Markdown("# AccDiffusion")
             prompt = gr.Textbox(label="Prompt")
-            submit_btn = gr.Button("SUbmit")
-            # output_images = gr.Image(format="png")
-            output_images = gr.Textbox()
+            submit_btn = gr.Button("Submit")
+            output_images = gr.Gallery(format="png")
         submit_btn.click(
             fn = infer,
             inputs = [prompt],