phyloforfun commited on
Commit
98a227a
1 Parent(s): ef4e9d6

fix safety check

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. vouchervision/utils_hf.py +1 -1
app.py CHANGED
@@ -216,10 +216,10 @@ if 'dir_images_local_TEMP' not in st.session_state:
216
  st.session_state['dir_images_local_TEMP'] = False
217
  if 'dir_uploaded_images' not in st.session_state:
218
  st.session_state['dir_uploaded_images'] = os.path.join(st.session_state.dir_home,'uploads')
219
- validate_dir(os.path.join(st.session_state.dir_home,'uploads'))
220
  if 'dir_uploaded_images_small' not in st.session_state:
221
  st.session_state['dir_uploaded_images_small'] = os.path.join(st.session_state.dir_home,'uploads_small')
222
- validate_dir(os.path.join(st.session_state.dir_home,'uploads_small'))
223
 
224
 
225
 
 
216
  st.session_state['dir_images_local_TEMP'] = False
217
  if 'dir_uploaded_images' not in st.session_state:
218
  st.session_state['dir_uploaded_images'] = os.path.join(st.session_state.dir_home,'uploads')
219
+ validate_dir(st.session_state['dir_uploaded_images'])
220
  if 'dir_uploaded_images_small' not in st.session_state:
221
  st.session_state['dir_uploaded_images_small'] = os.path.join(st.session_state.dir_home,'uploads_small')
222
+ validate_dir(st.session_state['dir_uploaded_images_small'])
223
 
224
 
225
 
vouchervision/utils_hf.py CHANGED
@@ -105,7 +105,7 @@ def save_uploaded_file(directory, img_file, image=None):
105
  # Return the full path of the saved image
106
  return full_path
107
  except:
108
- with Image.open(os.path.join(directory,img_file.name)) as image:
109
  full_path = os.path.join(directory, img_file.name)
110
  image.save(full_path, "JPEG")
111
  # Return the full path of the saved image
 
105
  # Return the full path of the saved image
106
  return full_path
107
  except:
108
+ with Image.open(img_file.name) as image:
109
  full_path = os.path.join(directory, img_file.name)
110
  image.save(full_path, "JPEG")
111
  # Return the full path of the saved image