Spaces:
Running
Running
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 |
+
|