Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
3db6ff1
1
Parent(s):
16b19d3
Update app
Browse files
app.py
CHANGED
|
@@ -906,7 +906,7 @@ def calc_att_map(vid_emb, aud_emb, model):
|
|
| 906 |
|
| 907 |
scores = model.logits_scale(scores[..., None]).squeeze(-1)
|
| 908 |
|
| 909 |
-
return scores.cpu().numpy()
|
| 910 |
|
| 911 |
def generate_video(frames, audio_file, video_fname):
|
| 912 |
|
|
|
|
| 906 |
|
| 907 |
scores = model.logits_scale(scores[..., None]).squeeze(-1)
|
| 908 |
|
| 909 |
+
return scores.detach().cpu().numpy()
|
| 910 |
|
| 911 |
def generate_video(frames, audio_file, video_fname):
|
| 912 |
|