[F] Fix examples

This commit is contained in:
2023-10-11 19:55:08 -04:00
parent b41935ff3c
commit bb3bfb4e5b
3 changed files with 6 additions and 6 deletions
+2 -2
View File
@@ -1,11 +1,11 @@
"""
This is an example using CLAPCAP for audio captioning.
"""
from CLAPWrapper import CLAPWrapper
from msclap import CLAP
# Load and initialize CLAP
weights_path = "weights_path"
clap_model = CLAPWrapper(weights_path, version = 'clapcap', use_cuda=False)
clap_model = CLAP(weights_path, version = 'clapcap', use_cuda=False)
#Load audio files
audio_files = ['audio_file']
+2 -2
View File
@@ -3,7 +3,7 @@ This is an example using CLAP to perform zeroshot
classification on ESC50 (https://github.com/karolpiczak/ESC-50).
"""
from CLAPWrapper import CLAPWrapper
from msclap import CLAP
from esc50_dataset import ESC50
import torch.nn.functional as F
import numpy as np
@@ -18,7 +18,7 @@ y = [prompt + x for x in dataset.classes]
# Load and initialize CLAP
weights_path = "weights_path"
clap_model = CLAPWrapper(weights_path, version = '2023', use_cuda=False)
clap_model = CLAP(weights_path, version = '2023', use_cuda=False)
# Computing text embeddings
text_embeddings = clap_model.get_text_embeddings(y)
+2 -2
View File
@@ -1,7 +1,7 @@
"""
This is an example using CLAP for zero-shot inference.
"""
from CLAPWrapper import CLAPWrapper
from msclap import CLAP
import torch.nn.functional as F
# Define classes for zero-shot
@@ -17,7 +17,7 @@ audio_files = ['audio_file']
# Load and initialize CLAP
weights_path = "weights_path"
# Setting use_cuda = True will load the model on a GPU using CUDA
clap_model = CLAPWrapper(weights_path, version = '2023', use_cuda=False)
clap_model = CLAP(weights_path, version = '2023', use_cuda=False)
# compute text embeddings from natural text
text_embeddings = clap_model.get_text_embeddings(class_prompts)