martin commited on
Commit
ed26be9
·
1 Parent(s): 67c46fd
Files changed (2) hide show
  1. Dockerfile +2 -2
  2. clone_hf_model.sh → start_app.sh +2 -0
Dockerfile CHANGED
@@ -41,6 +41,6 @@ RUN pip install -r requirements.txt
41
  COPY --chown=user . .
42
 
43
  RUN pip install gradio
44
- RUN chmod +x clone_hf_model.sh
45
  ENV HF_MODEL_PATH="/tmp/hf_model"
46
- CMD ["./clone_hf_model.sh", "$HF_MODEL_PATH", "&&", "python", "app.py", "--model", "$HF_MODEL_PATH"]
 
41
  COPY --chown=user . .
42
 
43
  RUN pip install gradio
44
+ RUN chmod +x start_app.sh
45
  ENV HF_MODEL_PATH="/tmp/hf_model"
46
+ CMD ["./start_app.sh", "$HF_MODEL_PATH"]
clone_hf_model.sh → start_app.sh RENAMED
@@ -55,3 +55,5 @@ for repo in "${REPOSITORIES[@]}"; do
55
  done
56
 
57
  echo "所有仓库已成功下载!"
 
 
 
55
  done
56
 
57
  echo "所有仓库已成功下载!"
58
+
59
+ python app.py --model $LOCAL_DIR