Update hifigan/inference_e2e.py

hifigan/inference_e2e.py

@@ -1,3 +1,26 @@
+from __future__ import absolute_import, division, print_function, unicode_literals
+
+import os
+import numpy as np
+import json
+import torch
+from scipy.io.wavfile import write
+from hifigan.env import AttrDict
+from hifigan.models import Generator
+from io import BytesIO
+
+h = None
+device = None
+
+
+def load_checkpoint(filepath, device):
+    assert os.path.isfile(filepath)
+    print("Loading '{}'".format(filepath))
+    checkpoint_dict = torch.load(filepath, map_location=device)
+    print("Complete.")
+    return checkpoint_dict
+
+
 def hifi_gan_inference(input_mel, checkpoint_file):
     print('Initializing Inference Process..')
     config_file = os.path.join(os.path.split(checkpoint_file)[0], 'config.json')
@@ -49,4 +72,5 @@ def hifi_gan_inference(input_mel, checkpoint_file):
     output_buffer = BytesIO()
     write(output_buffer, h.sampling_rate, audio)
     
-    return output_buffer.getvalue()
+    return output_buffer.getvalue()
+