Merge pull request #117 from jhj0517/colab-fix

modules/base_interface.py

@@ -15,7 +15,9 @@ class BaseInterface:
 
     @staticmethod
     def remove_input_files(file_paths: List[str]):
+        if not file_paths:
+            return
+
         for file_path in file_paths:
-            if file_path is None or not os.path.exists(file_path):
-                continue
-            os.remove(file_path)
+            if file_path and os.path.exists(file_path):
+                os.remove(file_path)

modules/faster_whisper_inference.py

@@ -124,7 +124,8 @@ class FasterWhisperInference(BaseInterface):
             print(f"Error transcribing file on line {e}")
         finally:
             self.release_cuda_memory()
-            self.remove_input_files([fileobj.name for fileobj in fileobjs])
+            if not fileobjs:
+                self.remove_input_files([fileobj.name for fileobj in fileobjs])
 
     def transcribe_youtube(self,
                            youtubelink: str,