(pdf_folder, model, processor)
| 175 | progress = tqdm(total=len(folders)) |
| 176 | |
| 177 | def process_folder(pdf_folder, model, processor): |
| 178 | source_text = open(f"{pdf_folder}/source.md").read() |
| 179 | bird_eye = json.load(open(f"{pdf_folder}/refined_doc.json")) |
| 180 | images = json.load(open(f"{pdf_folder}/image_caption.json")).keys() |
| 181 | output_dir = f"{pdf_folder}/docpres/{llm_name}" |
| 182 | if os.path.exists(output_dir + "/final.jsonl"): |
| 183 | progress.write(f"Skipping {pdf_folder}") |
| 184 | progress.update(1) |
| 185 | return |
| 186 | try: |
| 187 | generate_slides( |
| 188 | output_dir, |
| 189 | source_text, |
| 190 | bird_eye, |
| 191 | list(images), |
| 192 | model, |
| 193 | processor, |
| 194 | ) |
| 195 | progress.update(1) |
| 196 | except Exception as e: |
| 197 | print(f"Error in {pdf_folder}: {e}") |
| 198 | |
| 199 | # for folder in folders: |
| 200 | # process_folder(folder, model, processor) |
no test coverage detected