Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
+FROM python:3.10-slim
 
 WORKDIR /app
 
@@ -12,16 +12,15 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir \
+      "torch==2.3.0" \
+      torchvision \
       fastapi \
       uvicorn[standard] \
       "huggingface-hub>=0.23.2" \
       "transformers==4.44.2" \
       accelerate \
-      bitsandbytes \
-      torchvision \
       pillow \
-      requests \
-      flash-attn --no-build-isolation
+      requests
 
 COPY app.py /app/app.py