InventorsHub commited on
Commit
33c2a59
·
verified ·
1 Parent(s): e979f5e

Update speech_processing.py

Browse files
Files changed (1) hide show
  1. speech_processing.py +2 -2
speech_processing.py CHANGED
@@ -2,14 +2,14 @@ from transformers import SeamlessM4Tv2Model, AutoProcessor
2
  import numpy as np
3
  import torch
4
  from pydub import AudioSegment
5
- import spaces
6
  import functools
7
  # Load processor and model
8
  processor = AutoProcessor.from_pretrained("facebook/seamless-m4t-v2-large")
9
  model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large")
10
 
 
11
  @functools.lru_cache(maxsize=1)
12
- @spaces.CPU()
13
  def translate_audio(audio_file):
14
  if audio_file is None:
15
  return "No audio file detected. Please try again."
 
2
  import numpy as np
3
  import torch
4
  from pydub import AudioSegment
5
+ # import spaces
6
  import functools
7
  # Load processor and model
8
  processor = AutoProcessor.from_pretrained("facebook/seamless-m4t-v2-large")
9
  model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large")
10
 
11
+ # @spaces.CPU()
12
  @functools.lru_cache(maxsize=1)
 
13
  def translate_audio(audio_file):
14
  if audio_file is None:
15
  return "No audio file detected. Please try again."