phyloforfun commited on
Commit
ef4e9d6
1 Parent(s): 93ee5fc

fix safety check

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -272,7 +272,7 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
272
  # Determine the file type
273
  if uploaded_file.name.lower().endswith('.pdf'):
274
  # Handle PDF files
275
- file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file, img=None)
276
  # Convert each page of the PDF to an image
277
  n_pages = convert_pdf_to_jpg(file_path, st.session_state['dir_uploaded_images'], dpi=200)#st.session_state.config['leafmachine']['project']['dir_images_local'])
278
  # Update the input list for each page image
@@ -296,7 +296,7 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
296
  else:
297
  ind_small += 1
298
  # Handle JPG/JPEG files (existing process)
299
- file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file, img=None) ######### Yale TODO
300
  # file_path = os.path.join(st.session_state['dir_uploaded_images'], uploaded_file.name)
301
 
302
  st.session_state['input_list'].append(file_path)
 
272
  # Determine the file type
273
  if uploaded_file.name.lower().endswith('.pdf'):
274
  # Handle PDF files
275
+ file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file, image=None)
276
  # Convert each page of the PDF to an image
277
  n_pages = convert_pdf_to_jpg(file_path, st.session_state['dir_uploaded_images'], dpi=200)#st.session_state.config['leafmachine']['project']['dir_images_local'])
278
  # Update the input list for each page image
 
296
  else:
297
  ind_small += 1
298
  # Handle JPG/JPEG files (existing process)
299
+ file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file, image=None) ######### Yale TODO
300
  # file_path = os.path.join(st.session_state['dir_uploaded_images'], uploaded_file.name)
301
 
302
  st.session_state['input_list'].append(file_path)