changed dockerfile and requirements.txt

2026-05-11 11:38:40 +02:00
parent 5d3c22fd1c
commit c0bc9588f5
3 changed files with 14 additions and 3 deletions
@@ -6,11 +6,13 @@ RUN apt-get update && apt-get install -y \
    bash \
    && rm -rf /var/lib/apt/lists/*

-COPY . .
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+
+COPY requirements.txt .

 RUN pip install --no-cache-dir -r requirements.txt

-RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+Copy . .

 RUN chmod +x start_server.sh