muddasser commited on
Commit
7222f5f
·
verified ·
1 Parent(s): c553b62

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -2
Dockerfile CHANGED
@@ -35,6 +35,9 @@ COPY requirements.txt .
35
  RUN pip install --upgrade pip && \
36
  pip install --no-cache-dir -r requirements.txt
37
 
 
 
 
38
  # Copy the rest of the application
39
  COPY . .
40
 
@@ -44,7 +47,6 @@ ENV CHROMIUM_PATH=/usr/bin/chromium
44
  ENV DISPLAY=:99
45
  ENV PYTHONUNBUFFERED=1
46
  ENV STREAMLIT_OPT_OUT_OF_USAGE_STATISTICS=true
47
- ENV TRANSFORMERS_CACHE=/app/cache
48
  ENV HF_HOME=/app/cache
49
  ENV WDM_CACHE_DIR=/tmp/.wdm
50
  ENV CHROME_USER_DATA_DIR=/tmp/chrome-data
@@ -56,4 +58,4 @@ RUN chmod +x /usr/bin/chromedriver /usr/bin/chromium
56
  EXPOSE 8501
57
 
58
  # Run Streamlit
59
- CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.fileWatcherType=none", "--server.headless=true"]
 
35
  RUN pip install --upgrade pip && \
36
  pip install --no-cache-dir -r requirements.txt
37
 
38
+ # Verify transformers installation
39
+ RUN python -c "from transformers import pipeline; print('Transformers pipeline imported successfully')"
40
+
41
  # Copy the rest of the application
42
  COPY . .
43
 
 
47
  ENV DISPLAY=:99
48
  ENV PYTHONUNBUFFERED=1
49
  ENV STREAMLIT_OPT_OUT_OF_USAGE_STATISTICS=true
 
50
  ENV HF_HOME=/app/cache
51
  ENV WDM_CACHE_DIR=/tmp/.wdm
52
  ENV CHROME_USER_DATA_DIR=/tmp/chrome-data
 
58
  EXPOSE 8501
59
 
60
  # Run Streamlit
61
+ CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.fileWatcherType=none", "--server.headless=true"]