Ngoufack commited on
Commit
968d6cc
·
1 Parent(s): 5fd6be5
Files changed (1) hide show
  1. app.py +4 -4
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 = 16
22
- FILE_LIMIT_MB = 1000
23
  COMPUTE_TYPE = "float32"
24
  YT_LENGTH_LIMIT_S = 600 # limit to 1 hour YouTube files
25
 
26
  num_speakers = 2
27
- language = None
28
- model_size = 'large'
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):