Nick088 commited on
Commit
37f1dd0
·
verified ·
1 Parent(s): e264e3f

Update infer.py

Browse files
Files changed (1) hide show
  1. infer.py +2 -0
infer.py CHANGED
@@ -5,6 +5,8 @@ from RealESRGAN import RealESRGAN
5
  import tempfile
6
  import numpy as np
7
  import tqdm
 
 
8
 
9
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
10
 
 
5
  import tempfile
6
  import numpy as np
7
  import tqdm
8
+ import pydub
9
+ from pydub import AudioSegment
10
 
11
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
12