roychao19477 commited on
Commit
484532b
·
1 Parent(s): e3fdf8b

Update figs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -76,7 +76,7 @@ def enhance(filepath, model_name):
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
 
 
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.randn(segment_len - len(chunk), device=chunk.device) * 1e-4)
80
  chunk = torch.cat([chunk, pad])
81
  chunk = chunk.unsqueeze(0)
82