fix

Dockerfile

@@ -1,21 +1,12 @@
 # read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
-FROM python:3.9
-
-# WORKDIR /code
-# COPY ./requirements.txt /code/requirements.txt
-# RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-# COPY . .
-
+FROM python:3.9 AS downloader
+WORKDIR /app
 RUN mkdir -p ./models/7B
 RUN wget -O ./models/7B/gpt-j-ggml-model-q4_0.bin https://huggingface.co/ocordeiro/ggml-gpt-j-6b-q4_0/resolve/main/gpt-j-ggml-model-q4_0.bin
 
-
 FROM ghcr.io/ggerganov/llama.cpp:light
-
-# echo ls files
-RUN ls -la
-
-CMD ["-m", "./models/7B/gpt-j-ggml-model-q4_0.bin", "-p", "Building a website can be done in 10 simple steps:", "-n", "512"]
+COPY --from=downloader /app/models/7B/gpt-j-ggml-model-q4_0.bin ./models/7B/gpt-j-ggml-model-q4_0.bin
+CMD ["-m", "/models/7B/gpt-j-ggml-model-q4_0.bin", "-p", "Building a website can be done in 10 simple steps:", "-n", "512"]
 
 # CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]