gal save

app.py

@@ -243,7 +243,7 @@ def process_video_frames(
         )
         print_image_info(result_image)
         save_generated_image(result_image,index)
-        yield result_image
+        yield result_image,result_image
         processed_frames.append(result_image)
     
     yield processed_frames
@@ -540,7 +540,14 @@ def app_gradio():
                         # Photo Examples
                         root_path = "resource/demo/example"
                         with gr.Column():
-                            gal_output = gr.Gallery(label="Processed Frames")
+                            gal_output = gr.Gallery(
+                        label="Generated Images",
+                        show_label=True,
+                        elem_id="gal_output",
+                        columns=3,
+                        height=800,
+                        visible=True
+                    )
                             refresh_button = gr.Button("Refresh Gallery")
                         
 
@@ -551,7 +558,7 @@ def app_gradio():
                 refresh_button.click(
         fn=update_gallery,
         inputs=[],
-        outputs=[file_gallery],
+        outputs=[gal_output],
     )
 
                 submit_flux.click(