Update app.py

app.py

@@ -396,7 +396,7 @@ def reset_propagation(predictor, stored_inference_state):
     
     predictor.reset_state(stored_inference_state)
     print(f"RESET State: {stored_inference_state} ")
-    return stored_inference_state
+    return gr.update(value=None, visible=False), stored_inference_state, None
     
 with gr.Blocks() as demo:
     first_frame_path = gr.State()
@@ -541,7 +541,7 @@ with gr.Blocks() as demo:
     reset_prpgt_brn.click(
         fn = reset_propagation,
         inputs = [loaded_predictor, stored_inference_state],
-        outputs = [stored_inference_state],
+        outputs = [output_propagated, stored_inference_state, output_result],
         queue=False
     )