OpenSound commited on
Commit
33db348
·
verified ·
1 Parent(s): 56ec8ff

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -58,7 +58,7 @@ class FlexSED:
58
  clap_embeds = []
59
  with torch.no_grad():
60
  for event in events:
61
- text = f"The sound of {event.replace('_',' ')}"
62
  inputs = self.tokenizer([text], padding=True, return_tensors="pt")
63
  outputs = self.clap(**inputs)
64
  text_embeds = outputs.text_embeds.unsqueeze(1)
 
58
  clap_embeds = []
59
  with torch.no_grad():
60
  for event in events:
61
+ text = f"The sound of {event.replace('_', ' ').capitalize()}"
62
  inputs = self.tokenizer([text], padding=True, return_tensors="pt")
63
  outputs = self.clap(**inputs)
64
  text_embeds = outputs.text_embeds.unsqueeze(1)