add @space decorator to every function

modules/utils/subtitle_manager.py

@@ -1,5 +1,7 @@
 import re
 
+# Zero GPU
+import spaces
 
 def timeformat_srt(time):
     hours = time // 3600
@@ -117,7 +119,7 @@ def get_serialized_vtt(dicts):
         output += f'{dic["sentence"]}\n\n'
     return output
 
-
+@spaces.GPU(duration=120)
 def safe_filename(name):
     from app import _args
     INVALID_FILENAME_CHARS = r'[<>:"/\\|?*\x00-\x1f]'

modules/whisper/faster_whisper_inference.py

@@ -13,6 +13,7 @@ from argparse import Namespace
 from modules.whisper.whisper_parameter import *
 from modules.whisper.whisper_base import WhisperBase
 
+import spaces
 
 class FasterWhisperInference(WhisperBase):
     def __init__(self,
@@ -31,6 +32,7 @@ class FasterWhisperInference(WhisperBase):
         self.available_compute_types = self.get_available_compute_type()
         self.download_model(model_size="large-v2", model_dir=self.model_dir)
 
+    @spaces.GPU(duration=120)
     def transcribe(self,
                    audio: Union[str, BinaryIO, np.ndarray],
                    progress: gr.Progress,
@@ -90,6 +92,7 @@ class FasterWhisperInference(WhisperBase):
         print("transcribe: finished")
         return segments_result, elapsed_time
 
+    @spaces.GPU(duration=120)
     def update_model(self,
                      model_size: str,
                      compute_type: str,

modules/whisper/whisper_base.py

@@ -42,6 +42,7 @@ class WhisperBase(ABC):
         self.vad = SileroVAD()
 
     @abstractmethod
+    @spaces.GPU(duration=120)
     def transcribe(self,
                    audio: Union[str, BinaryIO, np.ndarray],
                    progress: gr.Progress,
@@ -50,6 +51,7 @@ class WhisperBase(ABC):
         pass
 
     @abstractmethod
+    @spaces.GPU(duration=120)
     def update_model(self,
                      model_size: str,
                      compute_type: str,
@@ -57,6 +59,7 @@ class WhisperBase(ABC):
                      ):
         pass
 
+    @spaces.GPU(duration=120)
     def run(self,
             audio: Union[str, BinaryIO, np.ndarray],
             progress: gr.Progress,
@@ -322,6 +325,7 @@ class WhisperBase(ABC):
                 pass
 
     @staticmethod
+    @spaces.GPU(duration=120)
     def generate_and_write_file(file_name: str,
                                 transcribed_segments: list,
                                 add_timestamp: bool,