Manjot Singh commited on
Commit
d390c52
·
1 Parent(s): 365179e

added import spaces fix

Browse files
Files changed (1) hide show
  1. audio_processing.py +2 -2
audio_processing.py CHANGED
@@ -18,7 +18,7 @@ import torch
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,7 +30,7 @@ def preprocess_audio(audio, chunk_size=CHUNK_LENGTH*16000, overlap=OVERLAP*16000
30
  chunks.append(chunk)
31
  return chunks
32
 
33
- +@spaces.GPU
34
  def process_audio(audio_file, translate=False, model_size="small"):
35
  start_time = time.time()
36
 
 
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
+ @spaces.GPU
34
  def process_audio(audio_file, translate=False, model_size="small"):
35
  start_time = time.time()
36