abcd-hide commited on
Commit
6056662
·
verified ·
1 Parent(s): 7e58dba

Upload 3 files

Browse files
Files changed (2) hide show
  1. app.py +2 -3
  2. requirements.txt +5 -6
app.py CHANGED
@@ -16,9 +16,8 @@ app.add_middleware(
16
  allow_headers=["*"],
17
  )
18
 
19
- # Pix2Textを初期化 (初回にモデルをダウンロードしてRAMに保持)
20
- # analyzerの無効化などは運用に合わせて調整可能
21
- p2t = Pix2Text()
22
 
23
  @app.get("/")
24
  def read_root():
 
16
  allow_headers=["*"],
17
  )
18
 
19
+ # Pix2Textを初期化 (高速トークナイザーのエラーを回避するため use_fast=False を指定)
20
+ p2t = Pix2Text(formula_config={'use_fast': False})
 
21
 
22
  @app.get("/")
23
  def read_root():
requirements.txt CHANGED
@@ -3,14 +3,13 @@ fastapi
3
  uvicorn
4
  python-multipart
5
  pillow
6
- torch==2.2.2+cpu
7
- torchvision==0.17.2+cpu
8
  --extra-index-url https://download.pytorch.org/whl/cpu
9
- transformers==4.37.2
10
- tokenizers==0.15.2
11
- huggingface-hub==0.23.2
12
  numpy<2.0.0
13
  onnxruntime==1.19.0
14
- optimum==1.17.1
15
  doclayout-yolo==0.0.4
16
  ultralytics==8.2.28
 
3
  uvicorn
4
  python-multipart
5
  pillow
6
+ torch==2.3.1+cpu
7
+ torchvision==0.18.1+cpu
8
  --extra-index-url https://download.pytorch.org/whl/cpu
9
+ transformers==4.44.2
10
+ huggingface-hub==0.24.5
 
11
  numpy<2.0.0
12
  onnxruntime==1.19.0
13
+ optimum==1.22.0
14
  doclayout-yolo==0.0.4
15
  ultralytics==8.2.28