jhj0517 commited on
Commit
306617d
·
1 Parent(s): bc29f8e

fix spaces bug

Browse files
modules/whisper/faster_whisper_inference.py CHANGED
@@ -30,6 +30,7 @@ class FasterWhisperInference(WhisperBase):
30
  self.device = self.get_device()
31
  self.available_models = self.model_paths.keys()
32
  self.available_compute_types = self.get_available_compute_type()
 
33
 
34
  @spaces.GPU(duration=120)
35
  def transcribe(self,
@@ -157,3 +158,12 @@ class FasterWhisperInference(WhisperBase):
157
  return "auto"
158
  else:
159
  return "cpu"
 
 
 
 
 
 
 
 
 
 
30
  self.device = self.get_device()
31
  self.available_models = self.model_paths.keys()
32
  self.available_compute_types = self.get_available_compute_type()
33
+ self.download_model(model_size="large-v2", model_dir=self.model_dir)
34
 
35
  @spaces.GPU(duration=120)
36
  def transcribe(self,
 
158
  return "auto"
159
  else:
160
  return "cpu"
161
+
162
+ @staticmethod
163
+ def download_model(model_size: str, model_dir: str):
164
+ faster_whisper.download_model(
165
+ size_or_id=model_size,
166
+ cache_dir=model_dir
167
+ )
168
+
169
+