Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.1-base-ubuntu22.04
+FROM python:3.10-slim
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE 1
@@ -10,9 +10,7 @@ ENV MPLCONFIGDIR=/tmp/matplotlib
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
-    python3 \
-    python3-pip \
-    python3-dev \
+    git \
     build-essential \
     libpoppler-cpp-dev \
     && rm -rf /var/lib/apt/lists/*
@@ -20,20 +18,19 @@ RUN apt-get update && apt-get install -y \
 # Create and set working directory
 WORKDIR /app
 
-# Install base packages first
-RUN pip install --no-cache-dir packaging setuptools wheel
-
-# Install PyTorch with CUDA 12.1 support
+# Install base packages
 RUN pip install --no-cache-dir \
-    torch==2.1.2+cu121 \
-    --extra-index-url https://download.pytorch.org/whl/cu121
+    packaging \
+    setuptools \
+    wheel \
+    numpy
 
-# Install vLLM and other packages in optimal order
+# Install PyTorch with CUDA support (Hugging Face Spaces provides CUDA)
 RUN pip install --no-cache-dir \
-    vllm==0.3.0 \
-    xformers==0.0.22.post7 \
-    sentence-transformers==2.2.2 \
-    flash-attn==2.3.3 --no-build-isolation
+    torch \
+    transformers \
+    sentence-transformers \
+    vllm
 
 # Create necessary directories
 RUN mkdir -p /data/hf_cache/txagent_models \