lpw commited on
Commit
73832d7
1 Parent(s): 0176a7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -11,15 +11,17 @@ 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
  TGT_LIST = ['en', 'fr', 'es']
16
  MODEL_LIST = ['xm_transformer_sm_all-en']
17
  for src in SRC_LIST:
18
  for tgt in TGT_LIST:
19
  if src != tgt:
20
  MODEL_LIST.append(f"textless_sm_{src}_{tgt}")
21
-
22
- MODEL_LIST.remove('textless_sm_it_fr') #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]).
 
23
  # 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]).
24
 
25
  examples = []
 
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:
19
  for tgt in TGT_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 = []