wsntxxn commited on
Commit
f4539fe
1 Parent(s): b65b329

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ import torchaudio
24
 
25
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
26
 
27
- # use the model trained on AudioCaps
28
  model = AutoModel.from_pretrained(
29
  "wsntxxn/effb2-trm-clotho-captioning",
30
  trust_remote_code=True
 
24
 
25
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
26
 
27
+ # use the model trained on Clotho
28
  model = AutoModel.from_pretrained(
29
  "wsntxxn/effb2-trm-clotho-captioning",
30
  trust_remote_code=True