Commit Graph
12 Commits
Author SHA1 Message Date
myhloli 012a46e07d refactor(magic-pdf): optimize model initialization and concurrency control
- Remove concurrency limit logic from app.py
- Update model initialization process in various modules
- Remove unused VRAM check for concurrency limit
- Refactor OCR model initialization in pdf_extract_kit.py
- Update txt_spans_extract_v2 function to use lang parameter instead of ocr_model
2024-12-06 20:35:43 +08:00
myhloli 104273cc79 fix(vram): improve VRAM checking logic
- Update VRAM checking logic in app.py and model_utils.py
- Add None and type checks for VRAM values
- Adjust concurrency limit calculation in app.py
- Modify clean_vram function to handle cases with no VRAM information
2024-12-03 14:22:35 +08:00
myhloli b1fe9d4f60 feat(gradio_app): implement dynamic concurrency limit based on VRAM
- Add get_concurrency_limit function to calculate concurrency limit based on VRAM
- Update clean_vram function and rename to get_vram for better clarity
- Apply concurrency limit to the to_markdown function in the Gradio app
2024-12-03 11:44:25 +08:00
icecraft fa3c453c93 fix: use concrete class instead of abstract class 2024-11-21 18:34:41 +08:00
icecraft 6a481320ea fix: using new data api replace old rw api 2024-11-18 17:03:53 +08:00
myhloli 8ea2381334 style(gradio-app): add missing file type in upload
- Add missing '.jpg' file type to the list of allowed file types for upload
2024-11-08 18:31:32 +08:00
myhloli 4834baf4ba feat: add support for non-PDF file conversion to PDF
- Implement to_pdf function to convert non-PDF files to PDF format
- Integrate file upload functionality for PDF and image files- Update UI to include file upload component and PDF preview
- Add conversion button and update its functionality to handle new file types
2024-10-23 19:29:31 +08:00
myhloli 1279f2cd0f feat(model): add support for DocLayout-YOLO model
- Add new layout model option: DocLayout-YOLO
- Implement model initialization and prediction for DocLayout-YOLO
- Update configuration options to include new model- Modify existing code to support both LayoutLMv3 and DocLayout-YOLO models
- Update Gradio app to support more Custom Switch
2024-10-23 17:07:39 +08:00
myhloli cf377ce4a1 update example files 2024-10-16 16:21:25 +08:00
myhloli a71db70314 feat: add arXiv paper link to header and adjust PDF parsing logic- Add arXiv paper link to the header template for easy access to the latest research paper.
- Modify the PDF parsing logic to handle edge cases more accurately, particularly in determining the number of lines in a block based on its height.
2024-10-08 11:35:04 +08:00
myhloli bb373980e2 feat(gradio_app): add examples accordion to the PDF conversion interface
Introduce an examples accordion within the Gradio application to provide users with
a selection of sample PDFs for demonstration purposes. The added functionality allows
users to quickly test the PDF conversion feature with various document types.
2024-09-12 11:38:32 +08:00
myhloli cf48c1f97e feat(gradio_app): add web app with PDF processing as a project 2024-09-09 18:07:15 +08:00