update docker

Dockerfile

@@ -1,15 +1,22 @@
-# CUDA-enabled base image
+# CUDA-enabled base image with Python 3.11 support
 FROM nvidia/cuda:11.8.0-base-ubuntu20.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 
-# Install required system packages
+# Install required system packages and Python 3.11
 RUN apt-get update && apt-get install -y \
-    python3 python3-pip git wget unzip && \
+    python3.11 python3.11-distutils python3.11-dev python3-pip git wget unzip && \
     rm -rf /var/lib/apt/lists/*
 
+# Update alternatives to ensure Python 3.11 is the default
+RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1 && \
+    update-alternatives --config python3
+
+# Upgrade pip to the latest version
+RUN python3 -m pip install --upgrade pip
+
 # Set the working directory
 WORKDIR /app
 

requirements.txt

@@ -1,11 +1,11 @@
-Flask==2.3.2
-transformers==4.27.4
-sentence-transformers==2.2.2
-torch==2.0.1
-torchvision==0.15.2
-faiss-gpu==1.7.2
-PyMuPDF==1.22.5
-python-pptx==0.6.21
-pandas==1.5.3
-numpy==1.24.2
+Flask
+transformers
+sentence-transformers
+torch
+torchvision
+faiss-cpu
+PyMuPDF
+python-pptx
+pandas
+numpy
 #huggingface-hub==0.15.1  # Ensure compatibility