Update Dockerfile

Dockerfile

@@ -28,7 +28,7 @@ WORKDIR $HOME/app
 RUN git clone -b magnet https://github.com/camenduru/audiocraft $HOME/app 
     
 # Install dependencies
-RUN pip install -q av julius omegaconf einops num2words flashy torch torchaudio torchmetrics sentencepiece gradio
+RUN pip install -q av julius omegaconf einops num2words flashy torchaudio==2.0.2 torchmetrics sentencepiece gradio
 RUN pip install -q https://github.com/camenduru/wheels/releases/download/colab/xformers-0.0.20+1dc3d7a.d20240115-cp310-cp310-linux_x86_64.whl
 RUN pip install -q .