Update bark_infer_cpu.py

bark_infer_cpu.py

@@ -1,19 +1,20 @@
 import argparse
 import torch
 import numpy as np
-from bark import generate_audio, preload_models
+from bark import generate_audio
 from scipy.io import wavfile
 import os
 
 # Tắt cảnh báo
 os.environ['BARK_IGNORE_WARNINGS'] = '1'
 
-# ✅ Bắt buộc thêm dòng này NGAY SAU import numpy và TRƯỚC preload_models()
-torch.serialization.add_safe_globals([np.core.multiarray.scalar])
+from bark.generation import preload_models
+import torch.serialization
 
 def safe_load_models():
     try:
-        preload_models()
+        with torch.serialization.safe_globals([np._core.multiarray.scalar]):
+            preload_models()
     except Exception as e:
         print(f"Model load error: {e}")
         raise