Update Dockerfile

Dockerfile

@@ -1,44 +1,31 @@
 FROM python:3.8-slim
 
-# Install system dependencies
+ENV PYTHONUNBUFFERED=1
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+
 RUN apt-get update && apt-get install -y \
-    wget \
-    git \
-    build-essential \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    libsm6 \
-    libxext6 \
-    libxrender-dev \
-    && rm -rf /var/lib/apt/lists/*
-
-# Install binvox for voxelization
-RUN wget -O /usr/local/bin/binvox https://www.patrickmin.com/binvox/linux64/binvox?rnd=16820668852988 \
-    && chmod +x /usr/local/bin/binvox
-
-# Set working directory
+    git wget libgl1-mesa-glx libglib2.0-0 unzip && \
+    rm -rf /var/lib/apt/lists/*
+
 WORKDIR /app
 
-# Clone RigNet repository
-RUN git clone https://github.com/zhan-xu/RigNet.git /app/rignet
+# Install binvox
+RUN wget -O binvox http://www.patrickmin.com/binvox/linux64/binvox && \
+    chmod +x binvox && mv binvox /usr/local/bin/
+
+# Clone and setup RigNet
+RUN git clone https://github.com/zhan-xu/RigNet.git rignet && \
+    cd rignet/checkpoints && \
+    wget 'https://umass-my.sharepoint.com/:u:/g/personal/zhanxu_umass_edu/EQAf8L0-F3BNny7F0wodqbABX3I_TTqJXCIGP1ESdEv-Hg?download=1' -O checkpoints.zip && \
+    unzip checkpoints.zip && rm checkpoints.zip
 
-# Copy requirements first for better caching
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir torch==1.12.0 torchvision==0.13.0 --index-url https://download.pytorch.org/whl/cpu && \
+    pip install --no-cache-dir -f https://data.pyg.org/whl/torch-1.12.0+cpu.html \
+    torch_scatter torch_sparse torch_cluster torch_spline_conv && \
+    pip install --no-cache-dir -r requirements.txt
 
-# Copy application file
 COPY app.py .
 
-# Verify checkpoints directory structure
-RUN ls -la Rignet/checkpoints/ || echo "Checkpoints need to be added manually"
-
-# Expose Gradio port
 EXPOSE 7860
-
-# Set environment variables
-ENV GRADIO_SERVER_NAME="0.0.0.0"
-ENV GRADIO_SERVER_PORT=7860
-ENV PYTHONPATH="Rignet:${PYTHONPATH}"
-
-# Run the application
 CMD ["python", "app.py"]