Update README.md
Browse files
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
|
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
|