fix spaces bug

modules/whisper/faster_whisper_inference.py

@@ -9,6 +9,7 @@ import ctranslate2
 import whisper
 import gradio as gr
 from argparse import Namespace
+import spaces
 
 from modules.whisper.whisper_parameter import *
 from modules.whisper.whisper_base import WhisperBase
@@ -30,6 +31,7 @@ class FasterWhisperInference(WhisperBase):
         self.available_models = self.model_paths.keys()
         self.available_compute_types = self.get_available_compute_type()
 
+    @spaces.GPU
     def transcribe(self,
                    audio: Union[str, BinaryIO, np.ndarray],
                    progress: gr.Progress,
@@ -87,6 +89,7 @@ class FasterWhisperInference(WhisperBase):
         elapsed_time = time.time() - start_time
         return segments_result, elapsed_time
 
+    @spaces.GPU
     def update_model(self,
                      model_size: str,
                      compute_type: str,
@@ -146,6 +149,7 @@ class FasterWhisperInference(WhisperBase):
         return ['int16', 'float32', 'int8', 'int8_float32']
 
     @staticmethod
+    @spaces.GPU
     def get_device():
         if torch.cuda.is_available():
             return "cuda"

modules/whisper/whisper_base.py

@@ -249,6 +249,7 @@ class WhisperBase(ABC):
             self.release_cuda_memory()
             self.remove_input_files([mic_audio])
 
+    @spaces.GPU
     def transcribe_youtube(self,
                            youtube_link: str,
                            file_format: str,