Update README.md

README.md

@@ -58,6 +58,20 @@ predictions = asr_classifier.classify_file("audio_file.wav", device)
 print(predictions)
 ```
 
+## Docker
+
+A Docker image for the model with a Gradio web interface is available here: https://hub.docker.com/repository/docker/porjaz/buki-wav2vec2-2.0/general
+To run the container with GPU, first pull the image:
+```
+docker pull porjaz/buki-wav2vec2-2.0
+```
+
+Then, run the container:
+```
+docker run -v models:/docker_wav2vec2_gradio_app/models -it --gpus all -p 7860:7860 porjaz/buki-wav2vec2-2.0
+```
+
+
 ## Training
 
 To fine-tune this model, you need to run: