Update Dockerfile

Dockerfile

@@ -1,18 +1,21 @@
-FROM nvidia/cuda:11.8.0-base-ubuntu22.04
+FROM python:3.10-slim
 
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y python3 python3-pip git curl && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
 
-RUN pip3 install --no-cache-dir \
-    torch torchvision torchaudio \
-    transformers fastapi uvicorn accelerate bitsandbytes
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-COPY app.py /app/app.py
-COPY requirements.txt /app/requirements.txt
+COPY app.py .
 
-ENV CUDA_VISIBLE_DEVICES=0
+RUN mkdir -p /tmp/transformers_cache && \
+    chmod 777 /tmp/transformers_cache
 
-EXPOSE 8000
+RUN useradd -m appuser
+USER appuser
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]