[feat] show memory usage

app.py

@@ -1,6 +1,7 @@
 # coding=utf-8
 import logging
 import sys
+import os
 logging.getLogger('numba').setLevel(logging.WARNING)
 
 logging.basicConfig(
@@ -61,6 +62,8 @@ def vits(text, language, speaker_id, noise_scale, noise_scale_w, length_scale):
         speaker_id = LongTensor([speaker_id]).to(device)
         audio = net_g_ms.infer(x_tst, x_tst_lengths, sid=speaker_id, noise_scale=noise_scale, noise_scale_w=noise_scale_w,
                                length_scale=length_scale)[0][0, 0].data.cpu().float().numpy()
+    with os.popen('free') as f:
+        logger.info(f.read())
 
     return "生成成功!", (22050, audio), f"生成耗时 {round(time.perf_counter()-start, 2)} s"