mckabue commited on
Commit
9082298
·
verified ·
1 Parent(s): 9c6eceb

RE_UPLOAD-REBUILD-RESTART

Browse files
Files changed (1) hide show
  1. app.py +19 -0
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)