shreekantkalwar commited on
Commit
c50a9eb
·
1 Parent(s): 0b684ab
Files changed (2) hide show
  1. Dockerfile +1 -0
  2. requirements.txt +0 -0
Dockerfile CHANGED
@@ -5,6 +5,7 @@ WORKDIR /app
5
  COPY . .
6
 
7
  RUN pip install --no-cache-dir -r requirements.txt
 
8
 
9
  # Create non-root user
10
  RUN useradd -m -u 1000 appuser
 
5
  COPY . .
6
 
7
  RUN pip install --no-cache-dir -r requirements.txt
8
+ RUN pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130
9
 
10
  # Create non-root user
11
  RUN useradd -m -u 1000 appuser
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ