jhj0517 commited on
Commit
204d276
1 Parent(s): 7515a26

fix NoneType error

Browse files
modules/faster_whisper_inference.py CHANGED
@@ -124,7 +124,8 @@ class FasterWhisperInference(BaseInterface):
124
  print(f"Error transcribing file on line {e}")
125
  finally:
126
  self.release_cuda_memory()
127
- self.remove_input_files([fileobj.name for fileobj in fileobjs])
 
128
 
129
  def transcribe_youtube(self,
130
  youtubelink: str,
 
124
  print(f"Error transcribing file on line {e}")
125
  finally:
126
  self.release_cuda_memory()
127
+ if not fileobjs:
128
+ self.remove_input_files([fileobj.name for fileobj in fileobjs])
129
 
130
  def transcribe_youtube(self,
131
  youtubelink: str,