Spaces:
Sleeping
Sleeping
NikitaSrivatsan
commited on
Commit
·
8971856
1
Parent(s):
01eda25
Fixed typo in checkpoint name
Browse files- audiocaptioner.py +1 -1
audiocaptioner.py
CHANGED
@@ -65,4 +65,4 @@ class AudioCaptioner(torch.nn.Module):
|
|
65 |
amodel = 'HTSAT-base'
|
66 |
)
|
67 |
with contextlib.redirect_stdout(io.StringIO()):
|
68 |
-
self.clap_model.load_ckpt(ckpt = '
|
|
|
65 |
amodel = 'HTSAT-base'
|
66 |
)
|
67 |
with contextlib.redirect_stdout(io.StringIO()):
|
68 |
+
self.clap_model.load_ckpt(ckpt = 'checkpoints/music_audioset_epoch_15_esc_90.14.pt')
|