Update webui.py

webui.py

@@ -33,7 +33,7 @@ def generate_clicked(task):
 
     with model_management.interrupt_processing_mutex:
         model_management.interrupt_processing = False
-    # outputs=[progress_html, progress_window, progress_gallery, gallery]
+    outputs=[progress_html, progress_window, progress_gallery, gallery]
     execution_start_time = time.perf_counter()
     finished = False
 
@@ -53,7 +53,7 @@ def generate_clicked(task):
                 # help bad internet connection by skipping duplicated preview
                 if len(task.yields) > 0:  # if we have the next item
                     if task.yields[0][0] == 'preview':   # if the next item is also a preview
-                        # print('Skipped one preview for better internet connection.')
+                        print('Skipped one preview for better internet connection.')
                         continue
 
                 percentage, title, image = product
@@ -96,7 +96,7 @@ shared.gradio_root = gr.Blocks(
     css=modules.html.css).queue()
 
 with shared.gradio_root:
-    currentTask = gr.State(worker.AsyncTask(args=[]))
+    currentTask = worker.AsyncTask(args=[])
     with gr.Row():
         with gr.Column(scale=2):
             with gr.Row():