Spaces:
Running
on
Zero
Running
on
Zero
Helw150
commited on
Commit
•
4898006
1
Parent(s):
0e3a570
Spaces GPU Move?
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ if gr.NO_RELOAD:
|
|
21 |
resampler = Audio(sampling_rate=16_000)
|
22 |
|
23 |
|
24 |
-
@spaces.GPU
|
25 |
@torch.no_grad
|
26 |
def diva_audio(audio_input, do_sample=False, temperature=0.001, prev_outs=None):
|
27 |
sr, y = audio_input
|
@@ -58,6 +58,7 @@ def process_audio(audio: tuple, state: AppState):
|
|
58 |
return audio, state
|
59 |
|
60 |
|
|
|
61 |
def response(state: AppState, audio: tuple):
|
62 |
if not audio:
|
63 |
return AppState()
|
|
|
21 |
resampler = Audio(sampling_rate=16_000)
|
22 |
|
23 |
|
24 |
+
@spaces.GPU
|
25 |
@torch.no_grad
|
26 |
def diva_audio(audio_input, do_sample=False, temperature=0.001, prev_outs=None):
|
27 |
sr, y = audio_input
|
|
|
58 |
return audio, state
|
59 |
|
60 |
|
61 |
+
@spaces.GPU(duration=40, progress=gr.Progress(track_tqdm=True))
|
62 |
def response(state: AppState, audio: tuple):
|
63 |
if not audio:
|
64 |
return AppState()
|