| import os | |
| os.system("python -m pip install --upgrade pip") | |
| os.system("python -m pip install PyMuPDF==1.23.26") | |
| os.system("python -m pip install torch==2.1.0") | |
| os.system("python -m pip install torchvision==0.16.0") | |
| os.system("python -m pip install 'git+https://github.com/facebookresearch/detectron2.git@898507047cf441a1e4be7a729270961c401c4354'") | |
| os.system("python -m pip install layoutparser==0.3.4 layoutparser[layoutmodels] layoutparser[ocr]") | |
| os.system("python -m pip install Pillow==9.5.0") | |
| os.system("python -m pip install imagehash==4.3.1") | |
| os.system("python -m pip install tensorflow==2.15.0") | |
| os.system("python -m pip install scikit-learn==1.3.2") | |
| from main import app | |
| model_path = './model/trained_model/model_final.pth' | |
| config_path = './model/trained_model/config.yaml' | |
| examples = [f'./demo-examples/{filename}' for filename in os.listdir('./demo-examples/')] | |
| app(model_path=model_path, config_path=config_path, examples=examples, debug=True) |