diff --git a/README.md b/README.md index 0626bbd..1f7ca11 100644 --- a/README.md +++ b/README.md @@ -111,6 +111,7 @@ pip install -r gradio_requirements.txt Run gradio server from terminal or [`gradio_app.ipynb`](./gradio_app.ipynb): ```bash python gradio_app.py +TMP_PATH=/tmp python gradio_app.py # if you want to change tmp folder path ``` It is ready to use on [default url](http://127.0.0.1:7860). diff --git a/gradio_app.py b/gradio_app.py index d14433f..32b3e72 100644 --- a/gradio_app.py +++ b/gradio_app.py @@ -15,7 +15,7 @@ import random import uuid -TMP_PATH = "./demo/temp" +TMP_PATH = os.getenv("TMP_PATH", "./demo/temp") device = "cuda" if torch.cuda.is_available() else "cpu" whisper_model, align_model, voicecraft_model = None, None, None