Update app

app.py

@@ -41,7 +41,7 @@ CHECKPOINT_PATH = "model_rgb.pth"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 use_cuda = torch.cuda.is_available()
 print("Use cuda status: ", use_cuda)
-batch_size = 12
+batch_size = 48
 fps = 25
 n_negative_samples = 100
 
@@ -1083,11 +1083,12 @@ def get_embeddings(video_sequences, audio_sequences, model, calc_aud_emb=True):
 		- audio_emb (array) : Audio embedding
 	'''
 
-	batch_size = 48
 	video_emb = []
 	audio_emb = []
 
-	for i in range(0, len(video_sequences), batch_size):
+	model = model.cuda()
+	
+	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 = torch.mean(vid_emb, axis=-1)