KunalThakare279 commited on
Commit
53a86f8
·
verified ·
1 Parent(s): d12f5b4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -18,6 +18,10 @@ USER root
18
  # Install Python dependencies
19
  RUN pip install --no-cache-dir -r requirements.txt
20
 
 
 
 
 
21
  RUN python -c "from transformers import pipeline; pipeline('text-generation', model='mistralai/Mistral-7B-v0.1', tokenizer='mistralai/Mistral-7B-v0.1')"
22
 
23
  # Set ownership of the application directory
 
18
  # Install Python dependencies
19
  RUN pip install --no-cache-dir -r requirements.txt
20
 
21
+ RUN pip install huggingface_hub
22
+
23
+ RUN huggingface-cli login --token $read
24
+
25
  RUN python -c "from transformers import pipeline; pipeline('text-generation', model='mistralai/Mistral-7B-v0.1', tokenizer='mistralai/Mistral-7B-v0.1')"
26
 
27
  # Set ownership of the application directory