Spaces:
Sleeping
Sleeping
Manjot Singh
commited on
Commit
·
365179e
1
Parent(s):
8041c04
added import spaces decroator
Browse files- app.py +2 -0
- audio_processing.py +2 -0
app.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
import gradio as gr
|
2 |
from audio_processing import process_audio, print_results
|
3 |
import torch
|
|
|
|
|
4 |
|
5 |
print(f"CUDA available: {torch.cuda.is_available()}")
|
6 |
if torch.cuda.is_available():
|
|
|
1 |
import gradio as gr
|
2 |
from audio_processing import process_audio, print_results
|
3 |
import torch
|
4 |
+
+import spaces
|
5 |
+
|
6 |
|
7 |
print(f"CUDA available: {torch.cuda.is_available()}")
|
8 |
if torch.cuda.is_available():
|
audio_processing.py
CHANGED
@@ -18,6 +18,7 @@ import torch
|
|
18 |
import numpy as np
|
19 |
logging.basicConfig(level=logging.INFO)
|
20 |
logger = logging.getLogger(__name__)
|
|
|
21 |
|
22 |
|
23 |
def preprocess_audio(audio, chunk_size=CHUNK_LENGTH*16000, overlap=OVERLAP*16000): # 2 seconds overlap
|
@@ -29,6 +30,7 @@ def preprocess_audio(audio, chunk_size=CHUNK_LENGTH*16000, overlap=OVERLAP*16000
|
|
29 |
chunks.append(chunk)
|
30 |
return chunks
|
31 |
|
|
|
32 |
def process_audio(audio_file, translate=False, model_size="small"):
|
33 |
start_time = time.time()
|
34 |
|
|
|
18 |
import numpy as np
|
19 |
logging.basicConfig(level=logging.INFO)
|
20 |
logger = logging.getLogger(__name__)
|
21 |
+
+import spaces
|
22 |
|
23 |
|
24 |
def preprocess_audio(audio, chunk_size=CHUNK_LENGTH*16000, overlap=OVERLAP*16000): # 2 seconds overlap
|
|
|
30 |
chunks.append(chunk)
|
31 |
return chunks
|
32 |
|
33 | |
34 |
def process_audio(audio_file, translate=False, model_size="small"):
|
35 |
start_time = time.time()
|
36 |
|