Update Dockerfile

Dockerfile

@@ -1,7 +1,6 @@
 FROM python:3.10-slim
 
 WORKDIR /app
-
 ENV HF_HOME=/tmp/huggingface
 ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
 ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
@@ -11,6 +10,7 @@ ENV MPLCONFIGDIR=/tmp/matplotlib
 ENV CUDA_VISIBLE_DEVICES=""
 ENV HOME=/tmp
 
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     build-essential \
@@ -25,12 +25,25 @@ RUN apt-get update && apt-get install -y \
     libopencv-dev \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt .
+# Install Point-E and dependencies
 RUN pip install --no-cache-dir -U pip && \
-    pip install --no-cache-dir -r requirements.txt && \
+    pip install --no-cache-dir torch==1.13.1 torchvision==0.14.1 --index-url https://download.pytorch.org/whl/cpu && \
+    pip install --no-cache-dir flask flask-cors open3d opencv-python-headless && \
+    pip install --no-cache-dir git+https://github.com/openai/point-e.git && \
     pip cache purge
 
+# Create a directory for model storage
+RUN mkdir -p /tmp/point_e_models
+
+# Copy the pre-trained model weight file if it's available in the build context
+# Otherwise, users will need to provide this file
+COPY base40M-textvec.pt /tmp/point_e_models/base40M-textvec.pt
+
+# Copy application code
 COPY app.py .
 
+# Expose the port for the Flask server
 EXPOSE 7860
-CMD ["python", "app.py"]
+
+# Command to run the application
+CMD ["python", "app.py"]