Return

app.py

@@ -989,6 +989,7 @@ def process_yield(input_image, final_prompt, generation_mode, n_prompt, randomiz
         time.sleep(1)
         yield (gr.update(), gr.update(), "In progress..." + str(i), gr.update(), gr.update(), gr.update(interactive=True))
     yield (gr.update(), gr.update(), "Finished", gr.update(), gr.update(), gr.update(interactive=False))
+    return (gr.update(), gr.update(), "Finished", gr.update(), gr.update(), gr.update(interactive=False))
 
 title_html = """
     <h1><center>FramePack</center></h1>