Spaces:
Running
on
Zero
Running
on
Zero
fix: add spaces dependencies for ZeroGPU
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ import uuid
|
|
| 7 |
from io import StringIO
|
| 8 |
|
| 9 |
import gradio as gr
|
|
|
|
| 10 |
import torch
|
| 11 |
import torchaudio
|
| 12 |
from huggingface_hub import HfApi, hf_hub_download, snapshot_download
|
|
@@ -15,7 +16,7 @@ from TTS.tts.models.xtts import Xtts
|
|
| 15 |
from vinorm import TTSnorm
|
| 16 |
|
| 17 |
# download for mecab
|
| 18 |
-
|
| 19 |
|
| 20 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 21 |
api = HfApi(token=HF_TOKEN)
|
|
@@ -88,6 +89,7 @@ def calculate_keep_len(text, lang):
|
|
| 88 |
return -1
|
| 89 |
|
| 90 |
|
|
|
|
| 91 |
def predict(
|
| 92 |
prompt,
|
| 93 |
language,
|
|
|
|
| 7 |
from io import StringIO
|
| 8 |
|
| 9 |
import gradio as gr
|
| 10 |
+
import spaces
|
| 11 |
import torch
|
| 12 |
import torchaudio
|
| 13 |
from huggingface_hub import HfApi, hf_hub_download, snapshot_download
|
|
|
|
| 16 |
from vinorm import TTSnorm
|
| 17 |
|
| 18 |
# download for mecab
|
| 19 |
+
os.system("python -m unidic download")
|
| 20 |
|
| 21 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 22 |
api = HfApi(token=HF_TOKEN)
|
|
|
|
| 89 |
return -1
|
| 90 |
|
| 91 |
|
| 92 |
+
@spaces.GPU
|
| 93 |
def predict(
|
| 94 |
prompt,
|
| 95 |
language,
|