Update Dockerfile

Dockerfile

@@ -1,6 +1,8 @@
-FROM python:latest
-
-ENV PYTHONUNBUFFERED 1
+FROM nvidia/cuda:11.6.0-base-ubuntu20.04
+RUN export PATH="/usr/local/cuda/bin:$PATH"
+RUN apt update && \
+    apt install --no-install-recommends -y build-essential python3 python3-pip wget && \
+    apt clean && rm -rf /var/lib/apt/lists/*
 
 EXPOSE 8000
 
@@ -15,7 +17,7 @@ RUN pip install --upgrade pip && \
 
 RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash
 RUN apt-get install git-lfs
-RUN git clone -b previous_llama_ggmlv2 https://huggingface.co/TheBloke/wizardLM-7B-GGML
+RUN git clone https://huggingface.co/TheBloke/wizardLM-7B-GGML
 
 COPY . .
 RUN ls -al