Update app

app.py

@@ -559,8 +559,7 @@ def load_checkpoint(path, model):
 		new_s[k.replace('module.', '')] = v
 	model.load_state_dict(new_s)
 
-	if use_cuda:
-		model.to(device)
+	model.to(device)
 
 	print("Loaded checkpoint from: {}".format(path))
 
@@ -1094,7 +1093,7 @@ def get_embeddings(video_sequences, audio_sequences, model, calc_aud_emb=True):
 	video_emb = []
 	audio_emb = []
 
-	model = model.to(device)
+	# model = model.to(device)
 
 	for i in tqdm(range(0, len(video_sequences), batch_size)):
 		video_inp = video_sequences[i:i+batch_size, ]  		
@@ -1110,6 +1109,8 @@ def get_embeddings(video_sequences, audio_sequences, model, calc_aud_emb=True):
 		
 		# torch.cuda.empty_cache()
 
+	print("Extracted embeddings")
+
 	video_emb = torch.cat(video_emb, dim=0)
 
 	if calc_aud_emb: