azalea/CLAP
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #28 from microsoft/bmartin1_update_examples

Browse Source 
Removed weights_path parameter
This commit is contained in:
Benjamin Elizalde
2023-10-20 11:33:18 -07:00
committed by GitHub
parent 8e87029c0e d32b134d27
commit 1af3d710ee
3 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/audio_captioning.py
+1 -2
View File
@@ -4,8 +4,7 @@ This is an example using CLAPCAP for audio captioning.
from msclap import CLAP
# Load and initialize CLAP
weights_path = "weights_path"
clap_model = CLAP(weights_path, version = 'clapcap', use_cuda=False)
clap_model = CLAP(version = 'clapcap', use_cuda=False)
#Load audio files
audio_files = ['audio_file']
examples/zero_shot_classification.py
+1 -2
View File
@@ -17,8 +17,7 @@ prompt = 'this is the sound of '
y = [prompt + x for x in dataset.classes]
# Load and initialize CLAP
weights_path = "weights_path"
clap_model = CLAP(weights_path, version = '2023', use_cuda=False)
clap_model = CLAP(version = '2023', use_cuda=False)
# Computing text embeddings
text_embeddings = clap_model.get_text_embeddings(y)
examples/zero_shot_predictions.py
+1 -2
View File
@@ -15,9 +15,8 @@ class_prompts = [prompt + x for x in classes]
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 = CLAP(weights_path, version = '2023', use_cuda=False)
clap_model = CLAP(version = '2023', use_cuda=False)
# compute text embeddings from natural text
text_embeddings = clap_model.get_text_embeddings(class_prompts)