Update app.py

app.py

@@ -252,7 +252,7 @@ def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_
     plt.savefig(output_filename, format='jpg')
     plt.close()
 
-    """
+    
     #### PROPAGATION ####
 
     # Define a directory to save the JPEG images
@@ -289,10 +289,10 @@ def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_
     
         # Close the plot
         plt.close()
-    """
+    
     # OLD
 
-    return output_filename
+    return output_filename, jpeg_images
 
 with gr.Blocks() as demo:
     first_frame_path = gr.State()
@@ -325,6 +325,7 @@ with gr.Blocks() as demo:
                 submit_btn = gr.Button("Submit")
             with gr.Column():
                 output_result = gr.Image()
+                output_propagated = gr.Gallery()
                 # output_result_mask = gr.Image()
     
     clear_points_btn.click(
@@ -351,7 +352,7 @@ with gr.Blocks() as demo:
     submit_btn.click(
         fn = sam_process,
         inputs = [input_first_frame_image, checkpoint, tracking_points, trackings_input_label, frames_output_dir],
-        outputs = [output_result]
+        outputs = [output_result, output_propagated]
     )
 
 demo.launch(show_api=False, show_error=True)