Tawhid Bin Omar commited on
Commit
8746a60
·
1 Parent(s): 159447d

Fix: Add protobuf dependency and remove problematic model pre-download

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -2
  2. requirements.txt +1 -0
Dockerfile CHANGED
@@ -13,9 +13,8 @@ COPY requirements.txt .
13
  # Install Python dependencies
14
  RUN pip install --no-cache-dir -r requirements.txt
15
 
16
- # Download models at build time
17
  RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')"
18
- RUN python -c "from transformers import pipeline; pipeline('text-classification', model='microsoft/deberta-v3-xsmall')"
19
 
20
  # Copy application code
21
  COPY . .
 
13
  # Install Python dependencies
14
  RUN pip install --no-cache-dir -r requirements.txt
15
 
16
+ # Pre-download embedding model (optional, saves first-request time)
17
  RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')"
 
18
 
19
  # Copy application code
20
  COPY . .
requirements.txt CHANGED
@@ -11,3 +11,4 @@ python-dotenv>=1.0.0
11
  huggingface-hub>=0.16.0
12
  requests>=2.31.0
13
  scikit-learn>=1.3.0
 
 
11
  huggingface-hub>=0.16.0
12
  requests>=2.31.0
13
  scikit-learn>=1.3.0
14
+ protobuf>=3.20.0