mac9087 commited on
Commit
9f6509a
·
verified ·
1 Parent(s): 138cb5e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -1
Dockerfile CHANGED
@@ -1,7 +1,7 @@
1
  # Use a lightweight Python image
2
  FROM python:3.10-slim
3
 
4
- # Set working direct
5
  WORKDIR /app
6
 
7
  # Set environment variables for Hugging Face cache
@@ -36,6 +36,11 @@ RUN pip install --no-cache-dir -U pip && \
36
  # Download U^2-Net model file from Google Drive
37
  RUN gdown https://drive.google.com/uc?id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ -O u2net.pth
38
 
 
 
 
 
 
39
  # Copy U^2-Net module and application code
40
  COPY u2net.py .
41
  COPY app.py .
 
1
  # Use a lightweight Python image
2
  FROM python:3.10-slim
3
 
4
+ # Set working directory
5
  WORKDIR /app
6
 
7
  # Set environment variables for Hugging Face cache
 
36
  # Download U^2-Net model file from Google Drive
37
  RUN gdown https://drive.google.com/uc?id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ -O u2net.pth
38
 
39
+ # Clone TripoSR repository and copy tsr module
40
+ RUN git clone https://github.com/Vast-AI-Research/TripoSR.git /tmp/TripoSR && \
41
+ cp -r /tmp/TripoSR/tsr /app/tsr && \
42
+ rm -rf /tmp/TripoSR
43
+
44
  # Copy U^2-Net module and application code
45
  COPY u2net.py .
46
  COPY app.py .