Update app

app.py

@@ -1093,18 +1093,18 @@ 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, ]  		
-		vid_emb = model.forward_vid(video_inp.to(device), return_feats=False)
+		vid_emb = model.forward_vid(video_inp, return_feats=False)
 		vid_emb = torch.mean(vid_emb, axis=-1)
 
 		video_emb.append(vid_emb.detach().cpu())
 
 		if calc_aud_emb:
 			audio_inp = audio_sequences[i:i+batch_size, ]
-			aud_emb = model.forward_aud(audio_inp.to(device))
+			aud_emb = model.forward_aud(audio_inp)
 			audio_emb.append(aud_emb.detach().cpu())
 		
 		# torch.cuda.empty_cache()