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 = 48
+batch_size = 24
 fps = 25
 n_negative_samples = 100
 
@@ -1321,9 +1321,9 @@ def process_video_syncoffset(video_path, num_avg_frames, apply_preprocess):
 		print("Successfully loaded the model")
 
 		# Extract embeddings
+		print("Obtaining audio and video embeddings...")
 		video_emb, audio_emb = get_embeddings(video_sequences, audio_sequences, model, calc_aud_emb=True)
 
-
 		# L2 normalize embeddings
 		video_emb = torch.nn.functional.normalize(video_emb, p=2, dim=1)
 		audio_emb = torch.nn.functional.normalize(audio_emb, p=2, dim=1)