|
|
|
FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04 |
|
|
|
|
|
WORKDIR /DocQA |
|
|
|
|
|
COPY ./requirements.txt ./ |
|
|
|
|
|
RUN apt-get update && apt-get install -y \ |
|
git \ |
|
python3.10 \ |
|
python3-pip \ |
|
&& apt-get clean |
|
RUN apt-get install poppler-utils -y |
|
RUN pip install --no-cache-dir -r requirements.txt |
|
RUN pip3 install torch --index-url https://download.pytorch.org/whl/cu121 |
|
|
|
|
|
COPY images ./images |
|
COPY app.py ./ |
|
COPY classication.py ./ |
|
COPY donut_inference.py ./ |
|
COPY non_form_llama_parse.py ./ |
|
COPY RAG.py ./ |
|
COPY best_resnet152_model.h5 ./ |
|
COPY Model ./ |
|
|
|
|
|
|
|
EXPOSE 8501 |
|
|
|
|
|
|
|
ENTRYPOINT ["streamlit", "run","app.py"] |
|
|