RE_UPLOAD-REBUILD-RESTART
Browse files
app.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import os
|
2 |
+
|
3 |
+
os.system("python -m pip install --upgrade pip")
|
4 |
+
os.system("python -m pip install PyMuPDF==1.23.26")
|
5 |
+
os.system("python -m pip install torch==2.1.0")
|
6 |
+
os.system("python -m pip install torchvision==0.16.0")
|
7 |
+
os.system("python -m pip install 'git+https://github.com/facebookresearch/detectron2.git@898507047cf441a1e4be7a729270961c401c4354'")
|
8 |
+
os.system("python -m pip install layoutparser==0.3.4 layoutparser[layoutmodels] layoutparser[ocr]")
|
9 |
+
os.system("python -m pip install Pillow==9.5.0")
|
10 |
+
os.system("python -m pip install imagehash==4.3.1")
|
11 |
+
os.system("python -m pip install tensorflow==2.15.0")
|
12 |
+
os.system("python -m pip install scikit-learn==1.3.2")
|
13 |
+
|
14 |
+
from main import app
|
15 |
+
|
16 |
+
model_path = './model/trained_model/model_final.pth'
|
17 |
+
config_path = './model/trained_model/config.yaml'
|
18 |
+
examples = [f'./demo-examples/{filename}' for filename in os.listdir('./demo-examples/')]
|
19 |
+
app(model_path=model_path, config_path=config_path, examples=examples, debug=True)
|