Update Dockerfile

Dockerfile

@@ -5,7 +5,7 @@ FROM python:3.9
 WORKDIR /app
 
 # 安装系统依赖
-RUN apt-get update && apt-get install -y curl sudo fontconfig
+RUN apt-get update && apt-get install -y curl sudo fontconfig wget
 
 # 设置所有环境变量到 /tmp 目录下
 ENV OLLAMA_HOME=/.ollama
@@ -32,12 +32,34 @@ RUN curl -L https://ollama.ai/install.sh | sh
 # 安装Gradio及其依赖
 RUN pip install gradio requests
 
-# 下载模型文件
-RUN curl -L https://huggingface.co/shenzhi-wang/Llama3.1-8B-Chinese-Chat/resolve/main/gguf/llama3.1_8b_chinese_chat_q4_0.gguf?download=true -o /app/llama3.1_8b_chinese_chat_q4_0.gguf
+# 下载模型文件（使用wget并添加重试机制）
+RUN wget --tries=3 --retry-connrefused --waitretry=5 --timeout=30 \
+    -O /app/model.gguf \
+    https://huggingface.co/shenzhi-wang/Llama3.1-8B-Chinese-Chat/resolve/main/gguf/llama3.1_8b_chinese_chat_q4_0.gguf
+
+# 验证文件是否为GGUF格式
+RUN python3 -c '
+import sys
+def is_gguf(file_path):
+    with open(file_path, "rb") as f:
+        magic = f.read(4)
+        return magic == b"GGUF"
+if not is_gguf("/app/model.gguf"):
+    print("Error: Downloaded file is not in GGUF format")
+    sys.exit(1)
+'
 
 # 将当前目录下的所有文件复制到容器内的/app目录
 COPY . /app
 
+# 创建新的Modelfile
+RUN echo "FROM /app/model.gguf\n\
+PARAMETER stop \"Human:\"\n\
+PARAMETER stop \"Assistant:\"\n\
+PARAMETER temperature 0.7\n\
+PARAMETER top_k 40\n\
+PARAMETER top_p 0.9" > /app/Modelfile
+
 # 添加启动脚本并设置执行权限
 COPY start.sh /app/start.sh
 RUN chmod +x /app/start.sh