Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.10-slim
+FROM python:3.10
 
 WORKDIR /app
 
@@ -11,22 +11,31 @@ RUN apt-get update && apt-get install -y \
     ffmpeg \
     libsm6 \
     libxext6 \
+    cmake \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first for better caching
-COPY requirements.txt .
+# Create output directory
+RUN mkdir -p outputs
+
+# Install PyTorch first separately (for better error reporting)
 RUN pip install --upgrade pip && \
-    pip install --no-cache-dir pyyaml && \
-    pip install --no-cache-dir -r requirements.txt
+    pip install --no-cache-dir torch==2.0.1 numpy==1.24.2
+
+# Install other Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Install Shap-E from source
+RUN pip install --no-cache-dir "git+https://github.com/openai/shap-e.git@main"
 
 # Copy the Flask application
 COPY app.py .
 
-# Create output directory
-RUN mkdir -p outputs
-
 # Expose the port
 EXPOSE 7860
 
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+
 # Run the application with timeout set to 300 seconds for longer model loading
 CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--timeout", "300", "app:app"]