Update extract_feature_print.py

extract_feature_print.py

@@ -157,9 +157,10 @@ def main():
 
     # Resolve model path and name
     custom_mappings = {
-        "wav2vec_2" : ("wav2vec_small_960h.pt", "wav2vec"),
+        "wav2vec" : ("wav2vec_small_960h.pt", "wav2vec"),
         "hubert_base": ("hubert_base.pt", "hubert"),
         "contentvec_base": ("contentvec_base.pt", "contentvec"),
+        "hubert_base_japanese" : ("hubert_base_japanese.pt","hubert_base_japanese")
     }
     if os.path.split(model_path)[-1] == "Custom" and model_name in custom_mappings:
         model_path, resolved_model_name = custom_mappings[model_name]