indahPurnamaSarii commited on
Commit
ae6ece7
·
1 Parent(s): f227f03

Pra-unduh model saat build untuk menghindari timeout

Browse files
Files changed (2) hide show
  1. Dockerfile +10 -6
  2. download_model.py +10 -0
Dockerfile CHANGED
@@ -1,8 +1,12 @@
1
- FROM python:3.9-slim
2
- WORKDIR /code
3
- COPY ./requirements.txt /code/requirements.txt
4
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 
 
 
 
 
5
  COPY . /code/
6
- RUN mkdir -p /app/.cache
7
- ENV SENTENCE_TRANSFORMERS_HOME=/app/.cache
8
- CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "app:app"]
 
1
+ # ... (baris FROM, WORKDIR, COPY requirements.txt) ...
2
+
 
3
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
4
+
5
+ # --- BARIS BARU UNTUK PRA-UNDUH MODEL ---
6
+ COPY ./download_model.py /code/download_model.py
7
+ RUN python /code/download_model.py
8
+ # --- AKHIR BARIS BARU ---
9
+
10
  COPY . /code/
11
+
12
+ # ... (sisa Dockerfile) ...
 
download_model.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ # download_model.py
2
+ from langchain_huggingface import HuggingFaceEmbeddings
3
+
4
+ print("Mulai pra-unduh model embedding...")
5
+
6
+ # Pastikan nama model ini sama dengan yang ada di app.py
7
+ # Jika Anda mengikuti saran saya sebelumnya, seharusnya 'all-MiniLM-L6-v2'
8
+ HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
9
+
10
+ print("Model embedding berhasil diunduh dan disimpan di cache.")