dingusagar commited on
Commit
9075b43
·
verified ·
1 Parent(s): eb36c8a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -36,6 +36,7 @@ RUN pip3 install ollama
36
  # Pull the required Ollama model
37
  #RUN ollama serve & sleep 5 && ollama pull llama3.2:1b
38
 
 
39
  RUN pip install gradio transformers
40
  # Copy your Python application
41
  COPY app.py $HOME/app/app.py
 
36
  # Pull the required Ollama model
37
  #RUN ollama serve & sleep 5 && ollama pull llama3.2:1b
38
 
39
+ RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
40
  RUN pip install gradio transformers
41
  # Copy your Python application
42
  COPY app.py $HOME/app/app.py