abcd-hide commited on
Commit
8f8a57e
·
verified ·
1 Parent(s): 59cbd73

Upload 2 files

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -1
  2. requirements.txt +7 -3
Dockerfile CHANGED
@@ -12,7 +12,8 @@ RUN apt-get update && apt-get install -y \
12
  && rm -rf /var/lib/apt/lists/*
13
 
14
  COPY requirements.txt .
15
- RUN pip install --no-cache-dir -r requirements.txt
 
16
 
17
  COPY . .
18
 
 
12
  && rm -rf /var/lib/apt/lists/*
13
 
14
  COPY requirements.txt .
15
+ RUN pip install --no-cache-dir --upgrade pip && \
16
+ pip install --no-cache-dir -r requirements.txt
17
 
18
  COPY . .
19
 
requirements.txt CHANGED
@@ -1,8 +1,12 @@
1
- pix2text
2
  fastapi
3
  uvicorn
4
  python-multipart
5
  pillow
6
- torch
7
- torchvision
8
  --extra-index-url https://download.pytorch.org/whl/cpu
 
 
 
 
 
1
+ pix2text==1.1.6
2
  fastapi
3
  uvicorn
4
  python-multipart
5
  pillow
6
+ torch==2.1.2+cpu
7
+ torchvision==0.16.2+cpu
8
  --extra-index-url https://download.pytorch.org/whl/cpu
9
+ transformers==4.37.2
10
+ huggingface-hub==0.20.3
11
+ onnxruntime==1.16.3
12
+ optimum==1.16.2