fix spaces bug

modules/diarize/diarizer.py

@@ -10,7 +10,6 @@ from modules.diarize.audio_loader import load_audio
 
 
 class Diarizer:
-    @spaces.GPU
     def __init__(self,
                  model_dir: str = os.path.join("models", "Diarization")
                  ):

modules/translation/nllb_inference.py

@@ -7,7 +7,6 @@ from modules.translation.translation_base import TranslationBase
 
 
 class NLLBInference(TranslationBase):
-    @spaces.GPU
     def __init__(self,
                  model_dir: str,
                  output_dir: str

modules/translation/translation_base.py

@@ -4,6 +4,7 @@ import gradio as gr
 from abc import ABC, abstractmethod
 from typing import List
 from datetime import datetime
+import spaces
 
 from modules.whisper.whisper_parameter import *
 from modules.utils.subtitle_manager import *
@@ -127,6 +128,7 @@ class TranslationBase(ABC):
             self.remove_input_files([fileobj.name for fileobj in fileobjs])
 
     @staticmethod
+    @spaces.GPU
     def get_device():
         if torch.cuda.is_available():
             return "cuda"
@@ -136,6 +138,7 @@ class TranslationBase(ABC):
             return "cpu"
 
     @staticmethod
+    @spaces.GPU
     def release_cuda_memory():
         if torch.cuda.is_available():
             torch.cuda.empty_cache()

modules/whisper/whisper_base.py

@@ -19,7 +19,6 @@ from modules.vad.silero_vad import SileroVAD
 
 
 class WhisperBase(ABC):
-    @spaces.GPU
     def __init__(self,
                  model_dir: str,
                  output_dir: str,
@@ -400,6 +399,7 @@ class WhisperBase(ABC):
         return time_str.strip()
 
     @staticmethod
+    @spaces.GPU
     def get_device():
         if torch.cuda.is_available():
             return "cuda"