Update gradio_app.py

This commit is contained in:
Rumah 2024-04-05 17:49:00 +08:00 committed by GitHub
parent b9ffda9405
commit 5c92b2864f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions

View File

@ -14,6 +14,8 @@ import numpy as np
import random import random
import uuid import uuid
os.chdir("/content/VoiceCraft-gradio-colab")
os.environ['USER'] = 'aaa'
TMP_PATH = os.getenv("TMP_PATH", "./demo/temp") TMP_PATH = os.getenv("TMP_PATH", "./demo/temp")
device = "cuda" if torch.cuda.is_available() else "cpu" device = "cuda" if torch.cuda.is_available() else "cpu"