Abhinav Gavireddi
commited on
Commit
·
8c61032
1
Parent(s):
33f4e34
fix: fixed issue with package versions
Browse files- .gitignore +3 -0
- Dockerfile +8 -1
- requirements.txt +0 -2
.gitignore
CHANGED
@@ -172,3 +172,6 @@ cython_debug/
|
|
172 |
|
173 |
# PyPI configuration file
|
174 |
.pypirc
|
|
|
|
|
|
|
|
172 |
|
173 |
# PyPI configuration file
|
174 |
.pypirc
|
175 |
+
|
176 |
+
# jupyter notebooks
|
177 |
+
*.ipynb
|
Dockerfile
CHANGED
@@ -12,15 +12,22 @@ RUN apt-get update && \
|
|
12 |
apt-get install -y --no-install-recommends \
|
13 |
build-essential \
|
14 |
ffmpeg \
|
15 |
-
|
16 |
# for hnswlib (needed for OpenMP)
|
17 |
libgomp1 \
|
|
|
|
|
18 |
&& rm -rf /var/lib/apt/lists/*
|
19 |
|
20 |
# Copy and install Python dependencies
|
21 |
COPY requirements.txt ./
|
22 |
RUN pip install --no-cache-dir -r requirements.txt
|
23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
# Copy application code
|
25 |
COPY src/ ./src/
|
26 |
COPY tests/ ./tests/
|
|
|
12 |
apt-get install -y --no-install-recommends \
|
13 |
build-essential \
|
14 |
ffmpeg \
|
|
|
15 |
# for hnswlib (needed for OpenMP)
|
16 |
libgomp1 \
|
17 |
+
curl \
|
18 |
+
git \
|
19 |
&& rm -rf /var/lib/apt/lists/*
|
20 |
|
21 |
# Copy and install Python dependencies
|
22 |
COPY requirements.txt ./
|
23 |
RUN pip install --no-cache-dir -r requirements.txt
|
24 |
|
25 |
+
# Install/patch MinerU and download models
|
26 |
+
RUN pip uninstall -y magic-pdf && \
|
27 |
+
pip install git+https://github.com/opendatalab/MinerU.git@dev && \
|
28 |
+
curl -L https://github.com/opendatalab/MinerU/raw/dev/scripts/download_models_hf.py -o download_models_hf.py && \
|
29 |
+
python download_models_hf.py
|
30 |
+
|
31 |
# Copy application code
|
32 |
COPY src/ ./src/
|
33 |
COPY tests/ ./tests/
|
requirements.txt
CHANGED
@@ -1,9 +1,7 @@
|
|
1 |
# Core
|
2 |
streamlit>=1.25.0
|
3 |
-
mineru>=0.1.0
|
4 |
sentence-transformers>=2.2.2
|
5 |
rank-bm25>=0.2.2
|
6 |
-
redis>=4.5.1
|
7 |
hnswlib>=0.7.0
|
8 |
transformers>=4.29.2
|
9 |
torch>=2.0.0
|
|
|
1 |
# Core
|
2 |
streamlit>=1.25.0
|
|
|
3 |
sentence-transformers>=2.2.2
|
4 |
rank-bm25>=0.2.2
|
|
|
5 |
hnswlib>=0.7.0
|
6 |
transformers>=4.29.2
|
7 |
torch>=2.0.0
|