Upload Dockerfile

Dockerfile

@@ -11,13 +11,9 @@ ENV HF_HOME=/code/cache
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN pip3 install --pre torch -f https://download.pytorch.org/whl/nightly/cu113/torch_nightly.html
+RUN pip3 install torch==1.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
 RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY ./main.py /code/app/main.py
 
-COPY ./data/kilt_wiki_prepared/ /code/data/kilt_wiki_prepared
-
-COPY ./data/kilt_wikipedia.faiss /code/data/kilt_wikipedia.faiss
-
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8080"]