diff --git a/Dockerfile b/Dockerfile
index 8931d1a..dcde342 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,9 +9,20 @@ WORKDIR /app
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN git clone https://github.com/facebookresearch/sscd-copy-detection.git
-RUN cd sscd-copy-detection && python -m pip install -r ./requirements.txt
-RUN mkdir models_files
-RUN cd sscd-copy-detection && wget https://dl.fbaipublicfiles.com/sscd-copy-detection/sscd_disc_mixup.torchscript.pt
+#RUN cd sscd-copy-detection && python -m pip install -r ./requirements.txt
+RUN pip install pytorch-lightning==1.6
+RUN pip install lightning-bolts==0.4.0
+RUN pip install classy_vision
+RUN pip install torch
+RUN pip install torchvision
+RUN pip install torchmetrics
+RUN pip install faiss-gpu
+RUN pip install augly
+RUN pip install pandas
+RUN pip install numpy
+RUN pip install tensorboard
+#RUN mkdir models_files
+#RUN cd sscd-copy-detection && wget https://dl.fbaipublicfiles.com/sscd-copy-detection/sscd_disc_mixup.torchscript.pt
 
 RUN apt-get update && apt-get install -y ffmpeg cmake swig libavcodec-dev libavformat-dev git
 RUN ln -s /usr/bin/ffmpeg /usr/local/bin/ffmpeg