Spaces:
Running
Running
Delete model_singleton.py
Browse files- model_singleton.py +0 -14
model_singleton.py
DELETED
@@ -1,14 +0,0 @@
|
|
1 |
-
import whisper
|
2 |
-
import torch
|
3 |
-
|
4 |
-
class WhisperModelSingleton:
|
5 |
-
_instance = None
|
6 |
-
_model = None
|
7 |
-
|
8 |
-
@staticmethod
|
9 |
-
def get_instance(model_name="whisper-large-v3"):
|
10 |
-
if WhisperModelSingleton._instance is None:
|
11 |
-
WhisperModelSingleton._instance = WhisperModelSingleton()
|
12 |
-
device = "cuda" if torch.cuda.is_available() else "cpu"
|
13 |
-
WhisperModelSingleton._model = whisper.load_model(model_name, device=device)
|
14 |
-
return WhisperModelSingleton._model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|