fix spaces bug

app.py

@@ -33,7 +33,6 @@ class App:
         whisper_type = self.args.whisper_type.lower().strip()
 
         if whisper_type in ["faster_whisper", "faster-whisper", "fasterwhisper"]:
-            print(f"\n\ndir: {self.args.faster_whisper_model_dir}\n\n")
             whisper_inf = FasterWhisperInference(
                 model_dir=self.args.faster_whisper_model_dir,
                 output_dir=self.args.output_dir,

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
@@ -141,6 +142,12 @@ class FasterWhisperInference(WhisperBase):
                 model_paths[model_name] = os.path.join(webui_dir, self.model_dir, model_name)
         return model_paths
 
+    @spaces.gpu
+    def get_available_compute_type(self):
+        if self.device == "cuda":
+            return ctranslate2.get_supported_compute_types("cuda")
+        return ctranslate2.get_supported_compute_types("cpu")
+
     @staticmethod
     def get_device():
         if torch.cuda.is_available():