Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +10 -11
Dockerfile
CHANGED
|
@@ -41,20 +41,19 @@ COPY --chown=1000:1000 data/ /app/data/
|
|
| 41 |
RUN ls -l /app/data/
|
| 42 |
|
| 43 |
# Verify data files or create chat_history.json if missing
|
| 44 |
-
RUN python -c "\
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
"
|
| 49 |
|
| 50 |
# Download models during build
|
| 51 |
RUN python -c "\
|
| 52 |
-
from sentence_transformers import SentenceTransformer; \
|
| 53 |
-
SentenceTransformer('all-MiniLM-L6-v2', cache_folder='/cache'); \
|
| 54 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer; \
|
| 55 |
-
AutoModelForCausalLM.from_pretrained('distilgpt2', cache_dir='/cache'); \
|
| 56 |
-
AutoTokenizer.from_pretrained('distilgpt2', cache_dir='/cache') \
|
| 57 |
-
"
|
| 58 |
|
| 59 |
# Set proper permissions
|
| 60 |
RUN chmod a+r /app/data/*.json && \
|
|
|
|
| 41 |
RUN ls -l /app/data/
|
| 42 |
|
| 43 |
# Verify data files or create chat_history.json if missing
|
| 44 |
+
RUN python -c "import os, json; \
|
| 45 |
+
assert os.path.exists('/app/data/rupeia_document.json'), 'Data file missing'; \
|
| 46 |
+
if not os.path.exists('/app/data/chat_history.json'): \
|
| 47 |
+
open('/app/data/chat_history.json', 'w').write('{}')"
|
|
|
|
| 48 |
|
| 49 |
# Download models during build
|
| 50 |
RUN python -c "\
|
| 51 |
+
from sentence_transformers import SentenceTransformer; \
|
| 52 |
+
SentenceTransformer('all-MiniLM-L6-v2', cache_folder='/cache'); \
|
| 53 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer; \
|
| 54 |
+
AutoModelForCausalLM.from_pretrained('distilgpt2', cache_dir='/cache'); \
|
| 55 |
+
AutoTokenizer.from_pretrained('distilgpt2', cache_dir='/cache') \
|
| 56 |
+
"
|
| 57 |
|
| 58 |
# Set proper permissions
|
| 59 |
RUN chmod a+r /app/data/*.json && \
|