tejani commited on
Commit
64f992e
·
verified ·
1 Parent(s): f8df845

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -1
Dockerfile CHANGED
@@ -12,7 +12,11 @@ WORKDIR /app
12
  RUN git clone https://github.com/rupeshs/fastsdcpu.git .
13
 
14
  # Install Python dependencies
15
- RUN pip install -r requirements.txt
 
 
 
 
16
 
17
  # Expose port for FastAPI server
18
  EXPOSE 8000
 
12
  RUN git clone https://github.com/rupeshs/fastsdcpu.git .
13
 
14
  # Install Python dependencies
15
+ RUN python -m venv env && \
16
+ . env/bin/activate && \
17
+ pip install --upgrade pip && \
18
+ pip install torch==2.2.2 --index-url https://download.pytorch.org/whl/cpu && \
19
+ pip install -r requirements.txt
20
 
21
  # Expose port for FastAPI server
22
  EXPOSE 8000