test propagate with return only

app.py

@@ -512,8 +512,8 @@ def on_image_click(
 @spaces.GPU()
 def propagate_masks(GLOBAL_STATE: gr.State):
     if GLOBAL_STATE is None or GLOBAL_STATE.inference_session is None:
-        yield "Load a video first.", gr.update()
-        return
+        # yield "Load a video first.", gr.update()
+        return GLOBAL_STATE, "Load a video first.", gr.update()
 
     processor = GLOBAL_STATE.processor
     model = GLOBAL_STATE.model
@@ -527,7 +527,7 @@ def propagate_masks(GLOBAL_STATE: gr.State):
     processed = 0
 
     # Initial status; no slider change yet
-    yield f"Propagating masks: {processed}/{total}", gr.update()
+    # yield f"Propagating masks: {processed}/{total}", gr.update()
 
     last_frame_idx = 0
     with torch.inference_mode():
@@ -550,10 +550,10 @@ def propagate_masks(GLOBAL_STATE: gr.State):
 
             processed += 1
             # Every 15th frame (or last), move slider to current frame to update preview via slider binding
-            if processed % 15 == 0 or processed == total:
-                yield f"Propagating masks: {processed}/{total}", gr.update(value=frame_idx)
-            else:
-                yield f"Propagating masks: {processed}/{total}", gr.update()
+            # if processed % 15 == 0 or processed == total:
+            #     yield f"Propagating masks: {processed}/{total}", gr.update(value=frame_idx)
+            # else:
+            #     yield f"Propagating masks: {processed}/{total}", gr.update()
 
     model.to("cpu")
     inference_session.inference_device = "cpu"
@@ -562,7 +562,12 @@ def propagate_masks(GLOBAL_STATE: gr.State):
     torch.cuda.empty_cache()
 
     # Final status; ensure slider points to last processed frame
-    yield (
+    # yield (
+    #     f"Propagated masks across {processed} frames for {len(inference_session.obj_ids)} objects.",
+    #     gr.update(value=last_frame_idx),
+    # )
+    return (
+        GLOBAL_STATE,
         f"Propagated masks across {processed} frames for {len(inference_session.obj_ids)} objects.",
         gr.update(value=last_frame_idx),
     )
@@ -826,7 +831,7 @@ with gr.Blocks(title="SAM2 Video (Transformers) - Interactive Segmentation", the
     propagate_btn.click(
         propagate_masks,
         inputs=[GLOBAL_STATE],
-        outputs=[propagate_status, frame_slider],
+        outputs=[GLOBAL_STATE, propagate_status, frame_slider],
     )
 
     reset_btn.click(