version upgrades

Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
 
 ENV NB_USER jovyan
 ENV NB_UID 1000
@@ -18,8 +18,8 @@ RUN export DEBIAN_FRONTEND=noninteractive \
     && add-apt-repository ppa:deadsnakes/ppa \
     && apt-get -yq update \
     && apt-get -yq install --no-install-recommends \
-        python3.10 \
-        python3.10-venv \
+        python3.11 \
+        python3.11-venv \
     &&  apt upgrade -y \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
@@ -34,7 +34,7 @@ WORKDIR /home/${NB_USER}/
 ENV PATH=/venv/bin:$PATH
 
 COPY --chown=${NB_USER}:users ./requirements.txt /home/${NB_USER}/requirements.txt
-RUN python3.10 -m venv /home/${NB_USER}/venv \
+RUN python3.11 -m venv /home/${NB_USER}/venv \
     && /home/${NB_USER}/venv/bin/pip install --upgrade pip wheel \
     && /home/${NB_USER}/venv/bin/pip install -r /home/${NB_USER}/requirements.txt
 

README.md

@@ -2,8 +2,8 @@
 title: Automatic speech recognition
 sdk: gradio
 app_file: src/app.py
-python_version: 3.10
-sdk_version: 4.21.0
+python_version: 3.11
+sdk_version: 4.40.0
 app_port: 7860
 tags: [asr, stt, speech-to-text, whisper, pyannote, diarization]
 pinned: true

requirements.txt

@@ -1,11 +1,11 @@
-transformers==4.38.2
-accelerate==0.27.2
-gradio==4.21.0
-pyannote.audio==3.1.1
-python-docx==0.8.11
-pandas==2.2.1
-python-dotenv
+transformers==4.43.4
+accelerate==0.33.0
+gradio==4.40.0
+pyannote.audio==3.3.1
+python-docx==1.1.2
+pandas==2.2.2
+python-dotenv==1.0.1
 optimum
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.1.2
-torchaudio==2.1.2
+torch==2.4.0
+torchaudio==2.4.0