Update app.py

app.py

@@ -11,15 +11,17 @@ description = "Gradio Demo for SpeechMatrix. To use it, simply record your audio
 
 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>"
 
-SRC_LIST = ['cs', 'de', 'en', 'es', 'et', 'fi', 'fr', 'hr', 'hu', 'it', 'nl', 'pl', 'pt', 'ro', 'sk', 'sl']
+# SRC_LIST = ['cs', 'de', 'en', 'es', 'et', 'fi', 'fr', 'hr', 'hu', 'it', 'nl', 'pl', 'pt', 'ro', 'sk', 'sl']
+SRC_LIST = ['cs', 'de', 'en', 'es', 'et', 'fi', 'fr', 'hr', 'hu', 'nl', 'pl', 'pt', 'ro', 'sk', 'sl']
 TGT_LIST = ['en', 'fr', 'es']
 MODEL_LIST = ['xm_transformer_sm_all-en']
 for src in SRC_LIST:
     for tgt in TGT_LIST:
         if src != tgt:
             MODEL_LIST.append(f"textless_sm_{src}_{tgt}")
-
-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]).
+            
+# for textless_sm_it_fr, textless_sm_it_es
+# 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]).
 # 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]).
 
 examples = []