don't use yield

app.py

@@ -509,7 +509,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 GLOBAL_STATE, "Load a video first.", gr.update()
+        # yield GLOBAL_STATE, "Load a video first.", gr.update()
+        return GLOBAL_STATE, "Load a video first.", gr.update()
 
     processor = deepcopy(GLOBAL_STATE.processor)
     model = deepcopy(GLOBAL_STATE.model)
@@ -523,7 +524,7 @@ def propagate_masks(GLOBAL_STATE: gr.State):
     processed = 0
 
     # Initial status; no slider change yet
-    yield GLOBAL_STATE, f"Propagating masks: {processed}/{total}", gr.update()
+    # yield GLOBAL_STATE, f"Propagating masks: {processed}/{total}", gr.update()
 
     last_frame_idx = 0
     with torch.inference_mode():
@@ -546,15 +547,16 @@ 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 GLOBAL_STATE, f"Propagating masks: {processed}/{total}", gr.update(value=frame_idx)
-            else:
-                yield GLOBAL_STATE, f"Propagating masks: {processed}/{total}", gr.update()
+            # if processed % 15 == 0 or processed == total:
+            #     yield GLOBAL_STATE, f"Propagating masks: {processed}/{total}", gr.update(value=frame_idx)
+            # else:
+            #     yield GLOBAL_STATE, f"Propagating masks: {processed}/{total}", gr.update()
 
     text = f"Propagated masks across {processed} frames for {len(inference_session.obj_ids)} objects."
 
     # Final status; ensure slider points to last processed frame
-    yield GLOBAL_STATE, text, gr.update(value=last_frame_idx)
+    # yield GLOBAL_STATE, text, gr.update(value=last_frame_idx)
+    return GLOBAL_STATE, text, gr.update(value=last_frame_idx)
 
 
 def reset_session(GLOBAL_STATE: gr.State) -> tuple[AppState, Image.Image, int, int, str]: