Spaces:
Running
on
Zero
Running
on
Zero
Trying to fix spaces
Browse files
app.py
CHANGED
|
@@ -36,7 +36,6 @@ diva_model = AutoModel.from_pretrained(
|
|
| 36 |
resampler = Audio(sampling_rate=16_000)
|
| 37 |
|
| 38 |
|
| 39 |
-
@spaces.GPU
|
| 40 |
@torch.no_grad
|
| 41 |
def diva_audio(audio_input, do_sample=False, temperature=0.001):
|
| 42 |
sr, y = audio_input
|
|
@@ -50,7 +49,7 @@ def diva_audio(audio_input, do_sample=False, temperature=0.001):
|
|
| 50 |
a["array"], None, do_sample=do_sample, max_new_tokens=256
|
| 51 |
)
|
| 52 |
|
| 53 |
-
|
| 54 |
def transcribe_wrapper(audio_input, state, model_order):
|
| 55 |
spinner = "◒"
|
| 56 |
d_resp = gr.Textbox(
|
|
@@ -70,8 +69,6 @@ def transcribe_wrapper(audio_input, state, model_order):
|
|
| 70 |
|
| 71 |
yield from transcribe(audio_input, state, model_order)
|
| 72 |
|
| 73 |
-
|
| 74 |
-
@spaces.GPU
|
| 75 |
def transcribe(audio_input, state, model_order):
|
| 76 |
if audio_input == None:
|
| 77 |
return (
|
|
|
|
| 36 |
resampler = Audio(sampling_rate=16_000)
|
| 37 |
|
| 38 |
|
|
|
|
| 39 |
@torch.no_grad
|
| 40 |
def diva_audio(audio_input, do_sample=False, temperature=0.001):
|
| 41 |
sr, y = audio_input
|
|
|
|
| 49 |
a["array"], None, do_sample=do_sample, max_new_tokens=256
|
| 50 |
)
|
| 51 |
|
| 52 |
+
@spaces.GPU()
|
| 53 |
def transcribe_wrapper(audio_input, state, model_order):
|
| 54 |
spinner = "◒"
|
| 55 |
d_resp = gr.Textbox(
|
|
|
|
| 69 |
|
| 70 |
yield from transcribe(audio_input, state, model_order)
|
| 71 |
|
|
|
|
|
|
|
| 72 |
def transcribe(audio_input, state, model_order):
|
| 73 |
if audio_input == None:
|
| 74 |
return (
|