jhj0517 commited on
Commit
b865fea
·
1 Parent(s): 3ea3fd0

debug decoration

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +3 -4
modules/whisper/whisper_base.py CHANGED
@@ -59,7 +59,8 @@ class WhisperBase(ABC):
59
  ):
60
  pass
61
 
62
- @spaces.GPU(duration=120)
 
63
  def run(self,
64
  audio: Union[str, BinaryIO, np.ndarray],
65
  *whisper_params,
@@ -160,13 +161,11 @@ class WhisperBase(ABC):
160
  try:
161
  files_info = {}
162
  for file in files:
163
- print("run started: ")
164
  transcribed_segments, time_for_task = self.run(
165
  file.name,
166
  *whisper_params,
167
  progress=progress
168
  )
169
- print("run finished: ")
170
 
171
  file_name, file_ext = os.path.splitext(os.path.basename(file.name))
172
  file_name = safe_filename(file_name)
@@ -196,7 +195,7 @@ class WhisperBase(ABC):
196
  except Exception as e:
197
  print(f"Error transcribing file: {e}")
198
  finally:
199
- #self.release_cuda_memory()
200
  if not files:
201
  self.remove_input_files([file.name for file in files])
202
 
 
59
  ):
60
  pass
61
 
62
+ # spaces is problematic
63
+ #@spaces.GPU(duration=120)
64
  def run(self,
65
  audio: Union[str, BinaryIO, np.ndarray],
66
  *whisper_params,
 
161
  try:
162
  files_info = {}
163
  for file in files:
 
164
  transcribed_segments, time_for_task = self.run(
165
  file.name,
166
  *whisper_params,
167
  progress=progress
168
  )
 
169
 
170
  file_name, file_ext = os.path.splitext(os.path.basename(file.name))
171
  file_name = safe_filename(file_name)
 
195
  except Exception as e:
196
  print(f"Error transcribing file: {e}")
197
  finally:
198
+ # self.release_cuda_memory()
199
  if not files:
200
  self.remove_input_files([file.name for file in files])
201