change
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
|
2 |
import warnings
|
3 |
-
|
4 |
warnings.filterwarnings("ignore", category=FutureWarning)
|
5 |
import logging
|
6 |
from argparse import ArgumentParser
|
@@ -119,7 +119,7 @@ def load_model_if_needed(
|
|
119 |
log.info(f"Model '{variant}' already loaded with current settings.")
|
120 |
return False
|
121 |
|
122 |
-
|
123 |
@torch.inference_mode()
|
124 |
def generate_audio_gradio(
|
125 |
prompt,
|
|
|
1 |
|
2 |
import warnings
|
3 |
+
import spaces
|
4 |
warnings.filterwarnings("ignore", category=FutureWarning)
|
5 |
import logging
|
6 |
from argparse import ArgumentParser
|
|
|
119 |
log.info(f"Model '{variant}' already loaded with current settings.")
|
120 |
return False
|
121 |
|
122 |
+
@spaces.GPU
|
123 |
@torch.inference_mode()
|
124 |
def generate_audio_gradio(
|
125 |
prompt,
|