Update Dockerfile

Dockerfile

@@ -2,8 +2,8 @@
 FROM python:3.10 as base
 
 # Set model
-ENV MODEL=mlabonne/UltraMerge-7B-GGUF
-ENV QUANT=Q6_K
+ENV MODEL=TheBloke/Yi-6B-200K-GGUF
+ENV QUANT=Q8_0
 ENV CHAT_TEMPLATE=chatml
 
 # Set the working directory
@@ -15,7 +15,7 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 # Download model
 RUN MODEL_NAME_FILE=$(echo ${MODEL#*/} | tr '[:upper:]' '[:lower:]' | sed 's/-gguf$//') && \
-    wget https://huggingface.co/mradermacher/UltraMerge-7B-GGUF/resolve/main/UltraMerge-7B.Q6_K.gguf -O model.gguf
+    wget https://huggingface.co/TheBloke/Yi-6B-200K-GGUF/resolve/main/yi-6b-200k.Q8_0.gguf -O model.gguf
 
 # Copy the rest of your application
 COPY . .