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([2, 120, 120, 3, 10])
-    train_audio_wave = pt.zeros([2, 261540])
-    train_audio_cnn = pt.zeros([2, 150, 512, 1])
+    train_visual = pt.zeros([1, 120, 120, 3, 10])
+    train_audio_wave = pt.zeros([1, 261540])
+    train_audio_cnn = pt.zeros([1, 150, 512, 1])
     
     mfcc = torchaudio.transforms.MFCC(n_mfcc=150, melkwargs={"n_fft": 1022, "n_mels": 150})