BramVanroy commited on
Commit
c02d909
β€’
1 Parent(s): 4f1e963

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -11
Dockerfile CHANGED
@@ -23,8 +23,6 @@ ENV HOME="/home/mateo_user"
23
  # Environment variables
24
  ENV PORT=7860 \
25
  SERVER="localhost" \
26
- BASE="" \
27
- DEMO_MODE="" \
28
  HF_HUB_ENABLE_HF_TRANSFER=1 \
29
  PATH="${HOME}/.local/bin:${PATH}" \
30
  USE_CUDA=false
@@ -50,16 +48,10 @@ RUN huggingface-cli download bert-base-multilingual-cased model.safetensors toke
50
  EXPOSE $PORT
51
 
52
  # Healthcheck to ensure the service is running
53
- HEALTHCHECK CMD curl --fail http://$SERVER:$PORT$BASE/_stcore/health || exit 1
54
 
55
  # Set the working directory to the Streamlit app
56
  WORKDIR src/mateo_st
57
 
58
- # Simplify the CMD script with conditional --use_cuda flag
59
- CMD streamlit run 01_🎈_MATEO.py \
60
- --server.port $PORT \
61
- --server.address $SERVER \
62
- $(if [ -n "$BASE" ]; then echo "--server.baseUrlPath $BASE"; fi) \
63
- $(if [ "$DEMO_MODE" = "true" ]; then echo "--server.maxUploadSize 1"; fi) \
64
- -- \
65
- $(if [ "$DEMO_MODE" = "true" ]; then echo "--demo_mode"; fi)
 
23
  # Environment variables
24
  ENV PORT=7860 \
25
  SERVER="localhost" \
 
 
26
  HF_HUB_ENABLE_HF_TRANSFER=1 \
27
  PATH="${HOME}/.local/bin:${PATH}" \
28
  USE_CUDA=false
 
48
  EXPOSE $PORT
49
 
50
  # Healthcheck to ensure the service is running
51
+ HEALTHCHECK CMD curl --fail http://$SERVER:$PORT/_stcore/health || exit 1
52
 
53
  # Set the working directory to the Streamlit app
54
  WORKDIR src/mateo_st
55
 
56
+ # Launch app
57
+ CMD ["streamlit", "run", "01_🎈_MATEO.py", "--server.port", "$PORT", "--server.enableXsrfProtection", "false", "--", "--no_cuda"]