Charles Kabui
gather package versions
05dbbf3
raw
history blame
707 Bytes
import os
os.system("apt-get install poppler-utils")
os.system("python -m pip install torch==2.1.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 tensorflow-estimator==2.15.0")
from main import app
model_path = './model/trained_model/model_final.pth'
config_path = './model/trained_model/config.yaml'
app(model_path=model_path, config_path=config_path, debug=True)