Update requirements.txt

requirements.txt

@@ -1,15 +1,35 @@
-fastapi
-uvicorn[standard]
-python-multipart
-pillow
-opencv-python-headless
-numpy==1.26.4
-torch==2.3.1
-torchvision==0.18.1
-scipy
-scikit-learn
-shapely
-alphashape
-pydantic
-onnxruntime==1.18.0
-onnx
+# ============================================
+# HUGGING FACE SPACES - DOCKER SDK GPU
+# CUDA 12.1 Compatible Packages
+# ============================================
+
+# FastAPI
+fastapi==0.109.0
+uvicorn[standard]==0.27.0
+python-multipart==0.0.6
+
+# Image Processing
+pillow==10.2.0
+opencv-python-headless==4.9.0.80
+numpy>=1.24.0,<2.0
+
+# PyTorch with CUDA 12.1 support
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.1.2+cu121
+torchvision==0.16.2+cu121
+
+# ONNX Runtime GPU - CUDA 12.x compatible
+onnxruntime-gpu==1.17.0
+onnx==1.15.0
+
+# CuPy for CUDA 12.x
+cupy-cuda12x==13.0.0
+
+# Scientific computing
+scipy==1.12.0
+scikit-learn==1.4.0
+shapely==2.0.2
+alphashape==1.3.1
+
+# Pydantic
+pydantic>=2.0