Liusuthu commited on
Commit
dc311f2
·
verified ·
1 Parent(s): cdc83c8

Update app_utils.py

Browse files
Files changed (1) hide show
  1. app_utils.py +3 -3
app_utils.py CHANGED
@@ -5,7 +5,7 @@ import mediapipe as mp
5
  from PIL import Image
6
  import cv2
7
  from pytorch_grad_cam.utils.image import show_cam_on_image
8
-
9
  # Importing necessary components for the Gradio app
10
  from model import pth_model_static, pth_model_dynamic, cam, pth_processing
11
  from face_utils import get_box, display_info
@@ -48,7 +48,7 @@ def text_api(text:str):
48
  def classify_continuous(audio):
49
  print(type(audio))
50
  print(audio)
51
- sample_rate, signal = AudioReader.read_wav_file(audio) # 这是语音的输入
52
  signal = signal.astype(np.float32)
53
  signal /= np.max(np.abs(signal))
54
  sf.write("data/a.wav", signal, sample_rate)
@@ -430,7 +430,7 @@ def video_test(video):
430
  my_audio_clip = AudioFileClip(video)
431
  my_audio_clip.write_audiofile("data/audio.wav",ffmpeg_params=["-ac","1"])
432
 
433
- audio="data/audio.wav"
434
  text,prob2,label,path=classify_continuous(audio)
435
 
436
  score2=10*prob2[0]-10*prob[1]
 
5
  from PIL import Image
6
  import cv2
7
  from pytorch_grad_cam.utils.image import show_cam_on_image
8
+ import scipy.io.wavfile as wav
9
  # Importing necessary components for the Gradio app
10
  from model import pth_model_static, pth_model_dynamic, cam, pth_processing
11
  from face_utils import get_box, display_info
 
48
  def classify_continuous(audio):
49
  print(type(audio))
50
  print(audio)
51
+ sample_rate, signal = (audio) # 这是语音的输入
52
  signal = signal.astype(np.float32)
53
  signal /= np.max(np.abs(signal))
54
  sf.write("data/a.wav", signal, sample_rate)
 
430
  my_audio_clip = AudioFileClip(video)
431
  my_audio_clip.write_audiofile("data/audio.wav",ffmpeg_params=["-ac","1"])
432
 
433
+ audio = wav.read('data/audio.wav')
434
  text,prob2,label,path=classify_continuous(audio)
435
 
436
  score2=10*prob2[0]-10*prob[1]