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

added import spaces decroator

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. 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