ayush2917 commited on
Commit
56e361f
·
verified ·
1 Parent(s): e937133

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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
- import os, json; \
46
- assert os.path.exists('/app/data/rupeia_document.json'), 'Data file missing'; \
47
- if not os.path.exists('/app/data/chat_history.json'): open('/app/data/chat_history.json', 'w').write('{}') \
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 && \