osanseviero
HF staff
commited on
Commit
010dad2
2 Parent(s): 3020f73 b184e9d

Merge branch 'main' of https://huggingface.co/osanseviero/hubert_asr_using_hub into main

Browse files
Files changed (2) hide show
  1. README.md +3 -1
  2. model.py +1 -0
README.md CHANGED
@@ -9,4 +9,6 @@ widget:
9
  src: https://cdn-media.huggingface.co/speech_samples/sample1.flac
10
  ---
11
 
12
- # Test for superb using hubert downstream ASR
 
 
9
  src: https://cdn-media.huggingface.co/speech_samples/sample1.flac
10
  ---
11
 
12
+ # Test for superb using hubert downstream ASR and upstream hubert model from the HF Hub
13
+
14
+ This repo uses: https://huggingface.co/osanseviero/hubert_base
model.py CHANGED
@@ -19,6 +19,7 @@ class PreTrainedModel(Runner):
19
  ckp["Args"].init_ckpt = ckp_file
20
  ckp["Args"].mode = "inference"
21
  ckp["Args"].upstream = "osanseviero/hubert_base"
 
22
  ckp["Args"].from_hf_hub = True
23
  ckp["Args"].device = "cpu" # Just to try in my computer
24
  ckp["Config"]["downstream_expert"]["datarc"]["dict_path"]=os.path.join(path,'char.dict')
19
  ckp["Args"].init_ckpt = ckp_file
20
  ckp["Args"].mode = "inference"
21
  ckp["Args"].upstream = "osanseviero/hubert_base"
22
+ ckp["Args"].upstream_model_name = "model.pt"
23
  ckp["Args"].from_hf_hub = True
24
  ckp["Args"].device = "cpu" # Just to try in my computer
25
  ckp["Config"]["downstream_expert"]["datarc"]["dict_path"]=os.path.join(path,'char.dict')