Spaces:
Running
Running
Commit
·
ef4e9d6
1
Parent(s):
93ee5fc
fix safety check
Browse files
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,
|
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,
|
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)
|