Merge pull request #42 from NourAlMerey/master

adding base image Dockerfile and updating README

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3
+
+# metainformation
+LABEL org.opencontainers.image.source = "https://github.com/Camb-ai/MARS5-TTS"
+LABEL org.opencontainers.image.licenses = "AGPL-3.0 license"
+
+
+# enable passwordless ssh
+RUN mkdir ~/.ssh && \
+    printf "Host * \n    ForwardAgent yes\nHost *\n    StrictHostKeyChecking no" > ~/.ssh/config && \
+    ssh-keygen -t rsa -N "" -f ~/.ssh/id_rsa && \
+    cat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys
+
+# enable RDMA support: this enables direct access to GPU memory
+RUN apt-get update && \
+    apt-get install -y infiniband-diags perftest ibverbs-providers libibumad3 libibverbs1 libnl-3-200 libnl-route-3-200 librdmacm1 && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# install MARS5-TTS
+RUN git clone https://github.com/Camb-ai/MARS5-TTS.git \
+    && cd ./MARS5-TTS \
+    && pip install -r requirements.txt

README.md

@@ -134,6 +134,24 @@ _Some tips for best quality:_
 - Use deep clone and provide an accurate transcript for the reference.
 - Use proper punctuation -- the model can be guided and made better or worse with proper use of punctuation and capitalization.
 
+## Or Use Docker
+
+**Pull from DockerHub**
+
+You can directly pull the docker image from our [DockerHub page](https://hub.docker.com/r/cambai/mars5ttsimage).
+
+
+**Build On Your Own**
+
+You can build a custom image from the provided Dockerfile in this repo by running the following command.
+
+```bash
+cd MARS5-TTS
+docker build -t mars5ttsimage ./docker
+```
+
+
+*Note: This image should be used as a base image on top of which you can add your custom inference script in a Dockerfile or docker-compose. Images that directly generate output will be added to Docker Hub and as Dockerfiles in this repo soon*
 
 ## Model Details