Spaces:
Sleeping
Sleeping
NikitaSrivatsan
commited on
Commit
·
01eda25
1
Parent(s):
6a4daf6
Added local version of CLAP checkpoint
Browse files- audiocaptioner.py +1 -1
- checkpoints/music_audioset_epoch_15_esc_90.14.pt +3 -0
- infer.py +0 -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')
|
checkpoints/music_audioset_epoch_15_esc_90.14.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fae3e9c087f2909c28a09dc31c8dfcdacbc42ba44c70e972b58c1bd1caf6dedd
|
3 |
+
size 2352471003
|
infer.py
CHANGED
@@ -18,7 +18,6 @@ def infer(input_filename):
|
|
18 |
print(f'Loaded from {checkpoint}')
|
19 |
model.eval()
|
20 |
# read in the wav file and precompute neighbors
|
21 |
-
#dataset_path = '/graft1/datasets/kechen/audiostock-full'
|
22 |
dataset_path = ''
|
23 |
train_dataset = AudiostockDataset(
|
24 |
dataset_path=dataset_path,
|
|
|
18 |
print(f'Loaded from {checkpoint}')
|
19 |
model.eval()
|
20 |
# read in the wav file and precompute neighbors
|
|
|
21 |
dataset_path = ''
|
22 |
train_dataset = AudiostockDataset(
|
23 |
dataset_path=dataset_path,
|