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)
|