sindhuhegde commited on
Commit
914fc28
·
1 Parent(s): 129cbf8

Update app

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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