JohnDoee commited on
Commit
366c5da
·
1 Parent(s): 2a34452

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -5
Dockerfile CHANGED
@@ -21,11 +21,7 @@ RUN python -m venv /app/venv && \
21
 
22
  # Ensure the model cache directory exists and is writable
23
  RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
24
-
25
- # Pre-download models
26
- RUN /app/venv/bin/python -c "from transformers import pipeline; \
27
- pipeline('sentiment-analysis', model='johndoee/sentiment')" || echo "Failed to download custom model"
28
-
29
  RUN /app/venv/bin/python -c "from transformers import pipeline; \
30
  pipeline('sentiment-analysis', model='tabularisai/multilingual-sentiment-analysis')" || echo "Failed to download model 1"
31
 
 
21
 
22
  # Ensure the model cache directory exists and is writable
23
  RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
24
+ =
 
 
 
 
25
  RUN /app/venv/bin/python -c "from transformers import pipeline; \
26
  pipeline('sentiment-analysis', model='tabularisai/multilingual-sentiment-analysis')" || echo "Failed to download model 1"
27