fix spaces bug

app.py

@@ -15,7 +15,11 @@ class App:
     def __init__(self, args):
         self.args = args
         self.app = gr.Blocks(css=CSS, theme=self.args.theme)
-        self.whisper_inf = self.init_whisper()
+        self.whisper_inf = FasterWhisperInference(
+            model_dir=self.args.faster_whisper_model_dir,
+            output_dir=self.args.output_dir,
+            args=self.args
+        )
         print(f"Use \"{self.args.whisper_type}\" implementation")
         print(f"Device \"{self.whisper_inf.device}\" is detected")
         self.nllb_inf = NLLBInference(

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,