Update app.py

app.py

@@ -3,16 +3,24 @@ from bark import SAMPLE_RATE, generate_audio, preload_models
 from scipy.io.wavfile import write as write_wav
 import tempfile
 import torch
-from numpy.core.multiarray import scalar
-import numpy
 
-# Add NumPy scalar and dtype to safe globals to fix UnpicklingError
-# This allowlists numpy.dtype to cover specific instances like numpy.dtype[float64]
-torch.serialization.add_safe_globals([scalar, numpy.dtype])
+# Save the original torch.load function
+original_load = torch.load
 
-# Preload the models at startup
+# Define a custom load function that forces weights_only=False
+def custom_load(*args, **kwargs):
+    kwargs['weights_only'] = False
+    return original_load(*args, **kwargs)
+
+# Monkey-patch torch.load with the custom function
+torch.load = custom_load
+
+# Preload the models with the patched torch.load
 preload_models()
 
+# Restore the original torch.load function
+torch.load = original_load
+
 def generate_speech(reference_audio, text):
     """
     Generate speech audio mimicking the voice from the reference audio using Bark.