Spaces:
Sleeping
Sleeping
unknown user
commited on
Commit
·
56a920f
1
Parent(s):
910f5b3
init
Browse files
app.py
CHANGED
@@ -3,9 +3,9 @@ from AinaTheme import theme
|
|
3 |
from faster_whisper import WhisperModel
|
4 |
import torch
|
5 |
|
6 |
-
device, torch_dtype = ("cuda", "float32") if torch.cuda.is_available() else ("cpu", "
|
7 |
|
8 |
-
MODEL_NAME = "
|
9 |
print("Loading model ...")
|
10 |
model = WhisperModel(MODEL_NAME, compute_type=torch_dtype)
|
11 |
print("Loading model done.")
|
|
|
3 |
from faster_whisper import WhisperModel
|
4 |
import torch
|
5 |
|
6 |
+
device, torch_dtype = ("cuda", "float32") if torch.cuda.is_available() else ("cpu", "float32")
|
7 |
|
8 |
+
MODEL_NAME = "suzii/vi-whisper-large-v3-turbo-v1-ct2"
|
9 |
print("Loading model ...")
|
10 |
model = WhisperModel(MODEL_NAME, compute_type=torch_dtype)
|
11 |
print("Loading model done.")
|