Update app.py

app.py

@@ -11,9 +11,9 @@ def process_video_audio(video_path, audio_path):
 
     wav = pt.tensor(list(audio_path[1]))
     
-    train_visual = np.zeros([5, 120, 120, 3, 10])
-    train_audio_wave = np.zeros([5, 261540])
-    train_audio_cnn = np.zeros([5, 150, 512, 1])
+    train_visual = np.zeros([1, 120, 120, 3, 10])
+    train_audio_wave = np.zeros([1, 261540])
+    train_audio_cnn = np.zeros([1, 150, 512, 1])
     
     mfcc = torchaudio.transforms.MFCC(n_mfcc=150, melkwargs={"n_fft": 1022, "n_mels": 150})
     
@@ -23,11 +23,7 @@ def process_video_audio(video_path, audio_path):
         train_audio_wave[0, :] = wav[:261540]
     else:
         train_audio_wave[0, :len(wav)] = wav[:]
-    
-    print(train_audio_wave.shape)
-    print(type(mfcc(train_audio_wave)))
-    
-    train_audio_cnn[:, :, :, 0] = mfcc(train_audio_wave)
+    train_audio_cnn[0, :, :, 0] = mfcc(train_audio_wave[0])
 
     print(train_audio_cnn[0].shape)