Spaces:
Running on Zero
Running on Zero
Commit ·
914fc28
1
Parent(s): 129cbf8
Update app
Browse files
app.py
CHANGED
|
@@ -1109,13 +1109,13 @@ def get_embeddings(video_sequences, audio_sequences, model, calc_aud_emb=True):
|
|
| 1109 |
|
| 1110 |
# torch.cuda.empty_cache()
|
| 1111 |
|
| 1112 |
-
print("Extracted embeddings")
|
| 1113 |
|
| 1114 |
# video_emb = torch.cat(video_emb, dim=0)
|
| 1115 |
|
| 1116 |
-
if calc_aud_emb:
|
| 1117 |
# audio_emb = torch.cat(audio_emb, dim=0)
|
| 1118 |
-
|
| 1119 |
return video_emb, audio_emb
|
| 1120 |
|
| 1121 |
return video_emb
|
|
|
|
| 1109 |
|
| 1110 |
# torch.cuda.empty_cache()
|
| 1111 |
|
| 1112 |
+
print("Extracted embeddings: ", len(video_emb), len(audio_emb))
|
| 1113 |
|
| 1114 |
# video_emb = torch.cat(video_emb, dim=0)
|
| 1115 |
|
| 1116 |
+
if calc_aud_emb==True:
|
| 1117 |
# audio_emb = torch.cat(audio_emb, dim=0)
|
| 1118 |
+
print("returning audio and video embeddings...")
|
| 1119 |
return video_emb, audio_emb
|
| 1120 |
|
| 1121 |
return video_emb
|