JotunnBurton commited on
Commit
36863d2
·
verified ·
1 Parent(s): a7af1b7

Update clap_wrapper.py

Browse files
Files changed (1) hide show
  1. clap_wrapper.py +2 -2
clap_wrapper.py CHANGED
@@ -9,8 +9,8 @@ from config import config
9
 
10
  models = dict()
11
 
12
- local_dir = snapshot_download(repo_id="laion/clap-htsat-fused", local_dir="./emotional/clap-htsat-fused")
13
- processor = ClapProcessor.from_pretrained(local_dir)
14
 
15
 
16
  def get_clap_audio_feature(audio_data, device=config.bert_gen_config.device):
 
9
 
10
  models = dict()
11
 
12
+ LOCAL_PATH = snapshot_download(repo_id="laion/clap-htsat-fused", local_dir="./emotional/clap-htsat-fused")
13
+ processor = ClapProcessor.from_pretrained(LOCAL_PATH)
14
 
15
 
16
  def get_clap_audio_feature(audio_data, device=config.bert_gen_config.device):