添加启动脚本和 HF_HOME 环境变量

.gitignore

@@ -1,2 +1,3 @@
 __pycache__
 /venv
+/huggingface

Dockerfile

@@ -6,8 +6,10 @@ WORKDIR /app
 
 COPY ./requirements_version.txt .
 
-RUN pip install -r requirements_version.txt --no-cache-dir
+RUN pip install -r requirements_version.txt --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple
 
 COPY . .
 
+RUN python acge_embedding.py
+
 CMD ["python", "main.py"]

start.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+
+export HF_HOME="$(pwd)/huggingface"
+echo HF_HOME is $HF_HOME
+source venv/bin/activate
+python3 main.py