Update app.py

app.py

@@ -90,7 +90,7 @@ def load_captioning(uploaded_files, concept_sentence):
     return updates
 
 def hide_captioning():
-    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) 
+    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) 
 
 def create_dataset(*inputs):
     print("Creating dataset")
@@ -473,7 +473,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
     )
     images.clear(
         hide_captioning,
-        outputs=[captioning_area, cost_preview, start]
+        outputs=[captioning_area, cost_preview, sample, start]
     )
     gr.on(
         triggers=[steps.change, payment_update.click],