THP2903 commited on
Commit
359e288
·
verified ·
1 Parent(s): 13ff32e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,11 +9,10 @@ emotion_labels = {0: 'neutral', 1: 'calm', 2: 'happy', 3: 'sad', 4: 'angry', 5:
9
 
10
  def process_video_audio(video_path, audio_path):
11
 
12
- wav = pt.tensor(audio_path[1])
13
 
14
  train_visual = np.zeros([1, 120, 120, 3, 10])
15
  train_audio_wave = np.zeros([1, 261540])
16
-
17
  train_audio_cnn = np.zeros([1, 150, 512, 1])
18
 
19
  mfcc = torchaudio.transforms.MFCC(n_mfcc=150, melkwargs={"n_fft": 1022, "n_mels": 150})
 
9
 
10
  def process_video_audio(video_path, audio_path):
11
 
12
+ wav = pt.tensor(audio_path[1], dtype = pt.float16)
13
 
14
  train_visual = np.zeros([1, 120, 120, 3, 10])
15
  train_audio_wave = np.zeros([1, 261540])
 
16
  train_audio_cnn = np.zeros([1, 150, 512, 1])
17
 
18
  mfcc = torchaudio.transforms.MFCC(n_mfcc=150, melkwargs={"n_fft": 1022, "n_mels": 150})