Update Dockerfile

Dockerfile

@@ -7,28 +7,29 @@ WORKDIR /app
 # Copy requirements
 COPY requirements.txt .
 
-# --- STABLE INSTALLATION BLOCK ---
-# 1. Install Torch 2.1.2 (Stable CPU version)
+# 1. Install specific NumPy version FIRST to prevent conflicts
+RUN pip install "numpy<2.0.0"
+
+# 2. Install Torch 2.1.2 (Stable CPU version)
 RUN pip install torch==2.1.2 --index-url https://download.pytorch.org/whl/cpu
 
-# 2. Install PyTorch Geometric Dependencies (Matching Torch 2.1.2)
+# 3. Install PyTorch Geometric Dependencies (Matching Torch 2.1.2)
 RUN pip install torch-scatter torch-sparse torch-cluster torch-spline-conv -f https://data.pyg.org/whl/torch-2.1.2+cpu.html
 
-# 3. Install PyTorch Geometric itself
+# 4. Install PyTorch Geometric
 RUN pip install torch-geometric
 
-# 4. Install the rest (FastAPI, etc.)
+# 5. Install the rest (FastAPI, scikit-learn, etc.)
 RUN pip install --no-cache-dir -r requirements.txt
-# -------------------------------
 
 # Copy the rest of the app
 COPY . .
 
-# Fix permissions for caching
+# Fix permissions
 RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
 
-# Expose the port
+# Expose port
 EXPOSE 7860
 
-# Start the app
+# Run
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]