diff --git a/gradio_app.py b/gradio_app.py index 9183e66..d0b442a 100644 --- a/gradio_app.py +++ b/gradio_app.py @@ -1,6 +1,4 @@ import os - os.chdir("/content/VoiceCraft-gradio-colab") - os.environ['USER'] = 'aaa' os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "0" # for local use import gradio as gr @@ -16,6 +14,9 @@ import numpy as np import random import uuid +os.chdir("/content/VoiceCraft-gradio-colab") +os.environ['USER'] = 'aaa' + 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