rodrigomasini commited on
Commit
af729d4
·
verified ·
1 Parent(s): c978fd7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -32,6 +32,8 @@ COPY requirements.txt .
32
  RUN python -m pip install --no-cache-dir --upgrade pip \
33
  && python -m pip install --no-cache-dir -r requirements.txt
34
 
 
 
35
  # Copy the application code into the container
36
  COPY mdr_pdf_parser.py .
37
  COPY main.py .
 
32
  RUN python -m pip install --no-cache-dir --upgrade pip \
33
  && python -m pip install --no-cache-dir -r requirements.txt
34
 
35
+ RUN python -c "import torch; print(f'PyTorch version during build: {torch.__version__}'); print(f'Has get_default_device: {hasattr(torch, \'get_default_device\')}')"
36
+
37
  # Copy the application code into the container
38
  COPY mdr_pdf_parser.py .
39
  COPY main.py .