Update Dockerfile

Dockerfile

@@ -17,28 +17,21 @@ RUN apt-get update && apt-get install -y \
 # 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 torch==2.0.1 numpy==1.24.2
-
-# Install Jupyter-related dependencies needed by Shap-E
-RUN pip install --no-cache-dir ipywidgets jupyter
-
-# Install other Python dependencies
+# Copy application code
+COPY app.py .
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
 
-# Install Shap-E from source with all dependencies
-RUN pip install --no-cache-dir "git+https://github.com/openai/shap-e.git@main"
-
-# Copy the Flask application
-COPY app.py .
+# Install Python dependencies - let Shap-E determine the correct versions
+RUN pip install --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt && \
+    pip install --no-cache-dir "git+https://github.com/openai/shap-e.git@main"
 
 # Expose the port
 EXPOSE 7860
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
+ENV SHAPEE_NO_INTERACTIVE=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"]