Update README.md

README.md

@@ -14,22 +14,22 @@ This project fine-tunes a TTS (Text-to-Speech) model using an mp3 file extracted
 - **Source Video**: [YouTube Video](https://www.youtube.com/watch?v=u6J20_Aem3Y)
 - **Training Audio**: The mp3 file used for training is included in the `files` directory.
 
-### Hugging Face Space
-The fine-tuning process is based on the Hugging Face Space found here:  
-[FineTune Xtts Space](https://huggingface.co/spaces/drewThomasson/FineTune_Xtts)
+### dockerimage
+Fine tuned with this docker image
+[FineTune Xtts Space](https://hub.docker.com/r/athomasson2/fine_tune_xtts)
 
 ### Docker Setup
 
 #### With GPU
 To run the training with GPU support:
 ```bash
-docker run -it -p 7860:7860 --gpus all --pull always --platform=linux/amd64 registry.hf.space/drewthomasson-finetune-xtts:latest python app.py
+docker run --gpus all -it -v ${PWD}\training:/tmp/xtts_ft/ athomasson2/fine_tune_xtts:v5
 ```
 
 #### Without GPU
 To run without GPU support:
 ```bash
-docker run -it -p 7860:7860 --pull always --platform=linux/amd64 registry.hf.space/drewthomasson-finetune-xtts:latest python app.py
+docker run -it -v ${PWD}/training:/tmp/xtts_ft/run/training athomasson2/fine_tune_xtts:v4_cpu
 ```
 
 ### Notes