casc
Browse files
app.py
CHANGED
@@ -18,14 +18,14 @@ from sklearn.cluster import AgglomerativeClustering
|
|
18 |
import numpy as np
|
19 |
|
20 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
21 |
-
BATCH_SIZE =
|
22 |
-
FILE_LIMIT_MB =
|
23 |
COMPUTE_TYPE = "float32"
|
24 |
YT_LENGTH_LIMIT_S = 600 # limit to 1 hour YouTube files
|
25 |
|
26 |
num_speakers = 2
|
27 |
-
language =
|
28 |
-
model_size = '
|
29 |
model_name = model_size
|
30 |
|
31 |
def getpreferredencoding(do_setlocale = True):
|
|
|
18 |
import numpy as np
|
19 |
|
20 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
21 |
+
BATCH_SIZE = 4
|
22 |
+
FILE_LIMIT_MB = 100
|
23 |
COMPUTE_TYPE = "float32"
|
24 |
YT_LENGTH_LIMIT_S = 600 # limit to 1 hour YouTube files
|
25 |
|
26 |
num_speakers = 2
|
27 |
+
language = "French"
|
28 |
+
model_size = 'medium'
|
29 |
model_name = model_size
|
30 |
|
31 |
def getpreferredencoding(do_setlocale = True):
|