Update app.py

app.py

@@ -265,7 +265,8 @@ if __name__ == "__main__":
                         interactive=True,
                     )
                     generate_mask_btn_video = gr.Button("1️⃣ Generate Mask", visible=True, variant="primary")
-                    gr.Examples([f"./demo/videos/{i+1}.mp4" for i in range(4)], inputs=video_input, label="Examples")
+                    qa_examples = gr.Examples([f"./demo/videos/{i+1}.mp4" for i in range(4)], inputs=video_input, label="Examples")
+                    seg_examples = gr.Examples([f"./demo/videos/{i+1}.mp4" for i in range(4,7)], inputs=video_input, label="Examples", visible=False)
 
                 with gr.Column():
                     mode_video = gr.Radio(label="Mode", choices=["QA", "Seg"], value="QA")
@@ -304,6 +305,8 @@ if __name__ == "__main__":
                 *[gr.update(visible=False) for _ in range(4)],
                 [],
                 "Please segment the rubbish bin.",
+                gr.update(visible=False),
+                gr.update(visible=True),
             )
 
         def to_qa_mode():
@@ -311,6 +314,8 @@ if __name__ == "__main__":
                 *[gr.update(visible=True) for _ in range(4)],
                 [],
                 "What's the function of <object0>?",
+                gr.update(visible=True),
+                gr.update(visible=False),
             )
 
         def on_mode_change(mode):
@@ -318,7 +323,7 @@ if __name__ == "__main__":
                 return to_qa_mode()
             return to_seg_mode()
 
-        mode_video.change(on_mode_change, inputs=[mode_video], outputs=[frame_idx, selected_frame, generate_mask_btn_video, response_video, mask_output_video, query_video])
+        mode_video.change(on_mode_change, inputs=[mode_video], outputs=[frame_idx, selected_frame, generate_mask_btn_video, response_video, mask_output_video, query_video, qa_examples, seg_examples])
         video_input.change(on_video_upload, inputs=[video_input, fps, max_frames], outputs=[frames, timestamps, selected_frame, frame_idx])
         frame_idx.change(on_frame_idx_change, inputs=[frame_idx, frames], outputs=[selected_frame])