diff --git a/.gitignore b/.gitignore index 3af6a7b..d3d8961 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ __pycache__ -/venv \ No newline at end of file +/venv +/huggingface diff --git a/Dockerfile b/Dockerfile index 8cf085f..fcd281b 100644 --- a/Dockerfile +++ b/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"] diff --git a/start.sh b/start.sh new file mode 100755 index 0000000..7a20935 --- /dev/null +++ b/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