Modify filepicker

file_picker.py

@@ -17,7 +17,8 @@ def update_dir(key):
     if os.path.isdir(os.path.join(st.session_state[key+'curr_dir'], choice)):
         st.session_state[key+'curr_dir'] = os.path.normpath(os.path.join(st.session_state[key+'curr_dir'], choice))
         files = sorted(os.listdir(st.session_state[key+'curr_dir']))
-        files.remove("images")
+        if "images" in files:
+          files.remove("images")
         st.session_state[key+'files'] = files
 
 def st_file_selector(st_placeholder, path='.', label='Select a file/folder', key = 'selected'):
@@ -28,7 +29,8 @@ def st_file_selector(st_placeholder, path='.', label='Select a file/folder', key
 
         files = sorted(os.listdir(base_path))
         files.insert(0, 'Choose a file...')
-        files.remove("images")
+        if "images" in files:
+          files.remove("images")
         st.session_state[key+'files'] = files
         st.session_state[key+'curr_dir'] = base_path
     else: