updating requirements

Author: taxe10

docker/Dockerfile renamed to Dockerfile

@@ -1,14 +1,12 @@
 FROM pytorch/pytorch:latest
 
-COPY docker/requirements.txt requirements.txt
+COPY requirements.txt requirements.txt
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
         tree
 RUN pip install --upgrade pip
-
 RUN pip install -r requirements.txt
 
 WORKDIR /app/work/
 COPY src/ src/
-
 CMD ["bash"]

docker/requirements.txt

@@ -0,0 +1,5 @@
+pandas==2.1.1
+Pillow==10.0.1
+pyarrow==13.0.0
+pydantic==1.8.2
+pytorch-lightning== 1.8.6

requirements.txt

@@ -1,13 +1,8 @@
-import os
-
-import einops
-from PIL import Image
-import numpy as np
 import pandas as pd
 import torch
-from torchvision import datasets, transforms
+from torchvision import transforms
 
-from model import CustomTensorDataset, CustomDirectoryDataset #, CustomSplashDataset
+from model import CustomDirectoryDataset
 
 
 def split_dataset(dataset, val_pct):