lpw commited on
Commit
682f715
1 Parent(s): 2f5bcc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -11,8 +11,8 @@ description = "Gradio Demo for SpeechMatrix. To use it, simply record your audio
11
 
12
  article = "<p style='text-align: center'><a href='https://research.facebook.com/publications/speechmatrix' target='_blank'>SpeechMatrix</a> | <a href='https://github.com/facebookresearch/fairseq/tree/ust' target='_blank'>Github Repo</a></p>"
13
 
14
- # SRC_LIST = ['cs', 'de', 'en', 'es', 'et', 'fi', 'fr', 'hr', 'hu', 'it', 'nl', 'pl', 'pt', 'ro', 'sk', 'sl']
15
- SRC_LIST = ['cs', 'de', 'en', 'es', 'et', 'fi', 'fr', 'hr', 'hu', 'nl', 'pl', 'pt', 'ro', 'sk', 'sl']
16
  TGT_LIST = ['en', 'fr', 'es']
17
  MODEL_LIST = ['xm_transformer_sm_all-en']
18
  for src in SRC_LIST:
@@ -20,10 +20,6 @@ for src in SRC_LIST:
20
  if src != tgt:
21
  MODEL_LIST.append(f"textless_sm_{src}_{tgt}")
22
 
23
- # for textless_sm_it_fr, textless_sm_it_es
24
- # size mismatch for source_unit_decoder.embed_tokens.weight: copying a param with shape torch.Size([804, 256]) from checkpoint, the shape in current model is torch.Size([1004, 256]).
25
- # size mismatch for source_unit_decoder.output_projection.weight: copying a param with shape torch.Size([804, 256]) from checkpoint, the shape in current model is torch.Size([1004, 256]).
26
-
27
  examples = []
28
  pipe_dict = {}
29
 
 
11
 
12
  article = "<p style='text-align: center'><a href='https://research.facebook.com/publications/speechmatrix' target='_blank'>SpeechMatrix</a> | <a href='https://github.com/facebookresearch/fairseq/tree/ust' target='_blank'>Github Repo</a></p>"
13
 
14
+ SRC_LIST = ['cs', 'de', 'en', 'es', 'et', 'fi', 'fr', 'hr', 'hu', 'it', 'nl', 'pl', 'pt', 'ro', 'sk', 'sl']
15
+ # SRC_LIST = ['cs', 'de', 'en', 'es', 'et', 'fi', 'fr', 'hr', 'hu', 'nl', 'pl', 'pt', 'ro', 'sk', 'sl']
16
  TGT_LIST = ['en', 'fr', 'es']
17
  MODEL_LIST = ['xm_transformer_sm_all-en']
18
  for src in SRC_LIST:
 
20
  if src != tgt:
21
  MODEL_LIST.append(f"textless_sm_{src}_{tgt}")
22
 
 
 
 
 
23
  examples = []
24
  pipe_dict = {}
25