Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
998d695
1
Parent(s):
4aa0f34
hotfix
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ import soundfile as sf
|
|
| 9 |
import torch
|
| 10 |
|
| 11 |
from dia.model import Dia
|
|
|
|
| 12 |
|
| 13 |
|
| 14 |
# Load Nari model and config
|
|
@@ -21,6 +22,7 @@ except Exception as e:
|
|
| 21 |
raise
|
| 22 |
|
| 23 |
|
|
|
|
| 24 |
def run_inference(
|
| 25 |
text_input: str,
|
| 26 |
audio_prompt_input: Optional[Tuple[int, np.ndarray]],
|
|
|
|
| 9 |
import torch
|
| 10 |
|
| 11 |
from dia.model import Dia
|
| 12 |
+
import spaces
|
| 13 |
|
| 14 |
|
| 15 |
# Load Nari model and config
|
|
|
|
| 22 |
raise
|
| 23 |
|
| 24 |
|
| 25 |
+
@spaces.GPU
|
| 26 |
def run_inference(
|
| 27 |
text_input: str,
|
| 28 |
audio_prompt_input: Optional[Tuple[int, np.ndarray]],
|