Update api.py
Browse files
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)
|