Fix Dockerfile for Hugging Face Spaces compatibility

Dockerfile

@@ -1,19 +1,4 @@
-FROM nvidia/cuda:11.8-devel-ubuntu22.04
-
-# Install Python 3.11
-RUN apt-get update && apt-get install -y \
-    software-properties-common \
-    && add-apt-repository ppa:deadsnakes/ppa \
-    && apt-get update && apt-get install -y \
-    python3.11 \
-    python3.11-pip \
-    python3.11-dev \
-    build-essential \
-    && rm -rf /var/lib/apt/lists/*
-
-# Set Python 3.11 as default
-RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1
-RUN update-alternatives --install /usr/bin/pip3 pip3 /usr/bin/pip3.11 1
+FROM python:3.11-slim
 
 # Set environment variables for GPU optimization
 ENV PYTHONUNBUFFERED=1
@@ -21,15 +6,21 @@ ENV PYTHONDONTWRITEBYTECODE=1
 ENV PIP_NO_CACHE_DIR=1
 ENV PIP_DISABLE_PIP_VERSION_CHECK=1
 ENV CUDA_VISIBLE_DEVICES=0
-ENV TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6"
 
 WORKDIR /code
 
+# Install system dependencies efficiently
+RUN apt-get update && apt-get install -y \
+    --no-install-recommends \
+    build-essential \
+    wget \
+    && rm -rf /var/lib/apt/lists/*
+
 # Copy and install requirements first (for better caching)
 COPY ./requirements.txt /code/requirements.txt
-RUN pip3 install --no-cache-dir --upgrade pip && \
-    pip3 install --no-cache-dir -r /code/requirements.txt && \
-    python3 -c "import nltk; nltk.download('punkt', quiet=True); nltk.download('stopwords', quiet=True)"
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r /code/requirements.txt && \
+    python -c "import nltk; nltk.download('punkt', quiet=True); nltk.download('stopwords', quiet=True)"
 
 # Copy application files
 COPY . .
@@ -41,5 +32,5 @@ RUN mkdir -p /.cache .chroma /root/.cache/huggingface && \
 # Expose port
 EXPOSE 7860
 
-# Use exec form with GPU-optimized settings
+# Use exec form with optimized settings
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]

requirements.txt

@@ -7,7 +7,8 @@ langchain-core==0.2.38
 langchain-openai==0.1.25
 rank_bm25==0.2.2
 transformers==4.42.0
-torch==2.1.0+cu118 --index-url https://download.pytorch.org/whl/cu118
+torch==2.1.0
+torchaudio==2.1.0
 accelerate==0.24.1
 sentence-transformers==2.2.2
 python-dotenv==1.0.0