Update README.md

README.md

@@ -78,8 +78,11 @@ emotion_label_list = [
     'Other'
 ]
     
-# Load data, here just zeros as the example, audio data should be 16kHz mono channel
-data = torch.zeros([1, 16000]).float().to(device)
+# Load data, here just zeros as the example
+# Our training data filters output audio shorter than 3 seconds (unreliable predictions) and longer than 15 seconds (computation limitation)
+# So you need to prepare your audio to a maximum of 15 seconds, 16kHz and mono channel
+max_audio_length = 15 * 16000
+data = torch.zeros([1, 16000]).float().to(device)[:, :max_audio_length]
 logits, embedding, _, _, _, _ = model(
     data, return_feature=True
 )