diff --git a/Dockerfile b/Dockerfile index 2af77ef..dfc1781 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM langchain/langchain +FROM pytorch/torchserve:latest-gpu # Seting the working directory WORKDIR /NeuarLearn-QA-ChatBot-RAG-Pipline