Rebuild Docker images for txtai 7.4, closes #14

Dockerfile

@@ -12,8 +12,8 @@ RUN \
     rm -rf /var/lib/apt/lists && \
     apt-get -y autoremove && \
     \
-    # Install base requirements, latest Transformers (Llama 3.1)
-    python -m pip install --no-cache-dir -r requirements.txt transformers --upgrade
+    # Install base requirements
+    python -m pip install --no-cache-dir -r requirements.txt
 
 # Start streamlit application
 ENTRYPOINT ["streamlit", "run", "rag.py"]

requirements.txt

@@ -1,4 +1,9 @@
+# Current version of AutoAWQ requires torch==2.3.1
 autoawq; platform_machine == "x86_64" or platform_machine == "AMD64"
+torch==2.3.1; platform_machine == "x86_64" or platform_machine == "AMD64"
+torchvision==0.18.1; platform_machine == "x86_64" or platform_machine == "AMD64"
+
+# Base requirements
 matplotlib
 streamlit
 txtai[graph,pipeline-data,pipeline-llm]