Correct Dockerfile and requirements

Author: victorjourne

Dockerfile

@@ -1,7 +1,7 @@
 FROM ovhcom/ai-training-pytorch:1.6.0
 LABEL maintainer="datalab-mi"
 
-COPY requirements.txt ./
-RUN pip install -r -y --no-cache-dir requirements.txt
+COPY requirements.txt requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
 

requirements.txt

@@ -2,4 +2,4 @@ pandas>=1.2.0
 requests>=2.21.0
 pytest==5.1.1
 transformers>=3.1.0,<5.0.0
-sentence-transformers=1.0.3
+sentence-transformers==1.0.3