roychao19477 commited on
Commit
6e97a1b
·
1 Parent(s): f9b4de2

Update chunk feature

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -112,7 +112,7 @@ def run_avse_inference(video_path, audio_path):
112
  # Version 1
113
  #estimated = avse_model.enhance(data).reshape(-1)
114
  # Version 2
115
- chunk_sec = 3
116
  sr = 16000
117
  audio_chunk_len = chunk_sec * sr # 48000
118
  video_chunk_len = chunk_sec * 25 # 75
@@ -137,10 +137,6 @@ def run_avse_inference(video_path, audio_path):
137
  "video_frames": video_chunk[np.newaxis, ...]
138
  }
139
 
140
- print(audio_chunk.shape)
141
- print(video_chunk.shape)
142
- fsefsef
143
-
144
  with torch.no_grad():
145
  out = avse_model.enhance(data).reshape(-1)
146
  estimated_chunks.append(out)
 
112
  # Version 1
113
  #estimated = avse_model.enhance(data).reshape(-1)
114
  # Version 2
115
+ chunk_sec = 6
116
  sr = 16000
117
  audio_chunk_len = chunk_sec * sr # 48000
118
  video_chunk_len = chunk_sec * 25 # 75
 
137
  "video_frames": video_chunk[np.newaxis, ...]
138
  }
139
 
 
 
 
 
140
  with torch.no_grad():
141
  out = avse_model.enhance(data).reshape(-1)
142
  estimated_chunks.append(out)