update

Dockerfile

@@ -1,16 +1,20 @@
+# Sử dụng một base image Alpine có Python 3.12 (nhẹ và ít lỗ hổng hơn)
 FROM python:3.12-alpine
 
 WORKDIR /app
 
-RUN apk add --no-cache \
-    build-base \
-    curl \
-    git
+# Cần cài đặt git nếu bạn có sử dụng nó trong ứng dụng (ví dụ: clone repo khác)
+# build-base thường không cần thiết trực tiếp nếu PyTorch được tải từ wheel
+RUN apk add --no-cache git curl
 
 COPY requirements.txt ./
 COPY src/ ./src/
 
-RUN pip3 install -r requirements.txt
+# !!! QUAN TRỌNG: SỬA DÒNG NÀY ĐỂ THÊM --index-url VÀ CHỌN PHIÊN BẢN CUDA CHÍNH XÁC !!!
+# Ví dụ với CUDA 11.8 (phổ biến trên các GPU cũ hơn như T4)
+RUN pip3 install --no-cache-dir -r requirements.txt --index-url https://download.pytorch.org/whl/cu118
+# Hoặc với CUDA 12.1 (hoặc cao hơn nếu GPU trên Space hỗ trợ và PyTorch cung cấp)
+# RUN pip3 install --no-cache-dir -r requirements.txt --index-url https://download.pytorch.org/whl/cu121
 
 EXPOSE 8501