roychao19477 commited on
Commit
77a842a
·
1 Parent(s): 776c247

Update figs

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -75,7 +75,8 @@ def enhance(filepath, model_name):
75
 
76
  for chunk in chunks:
77
  if len(chunk) < segment_len:
78
- pad = torch.zeros(segment_len - len(chunk), device=chunk.device)
 
79
  chunk = torch.cat([chunk, pad])
80
  chunk = chunk.unsqueeze(0)
81
 
 
75
 
76
  for chunk in chunks:
77
  if len(chunk) < segment_len:
78
+ #pad = torch.zeros(segment_len - len(chunk), device=chunk.device)
79
+ pad = torch.full(segment_len - len(chunk), 1e-4, device=chunk.device)
80
  chunk = torch.cat([chunk, pad])
81
  chunk = chunk.unsqueeze(0)
82