Update app.py

app.py

@@ -181,8 +181,8 @@ with gr.Blocks(css=css) as build:
                             save_btn=gr.Button("Protect")                            
                             rem_btn=gr.Button("Remove")                            
 
-    save_btn.click(add_save,[space_radio,save_list],save_list)
-    rem_btn.click(rem_save,[space_radio,save_list],save_list)
+    save_btn.click(add_save,[space_radio,save_list],save_list).then(show_s,[r_name,token,save_list],[space_radio])
+    rem_btn.click(rem_save,[space_radio,save_list],save_list).then(show_s,[r_name,token,save_list],[space_radio])
     
     move_btn.click(move,[r_name,space_radio,dest,token]).then(show_s,[r_name,token,save_list],[space_radio])
     s_btn.click(show_s,[r_name,token,save_list],[space_radio])