shukdevdatta123 commited on
Commit
c4c3acd
·
verified ·
1 Parent(s): 47d7c50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,9 +3,10 @@ from bark import SAMPLE_RATE, generate_audio, preload_models
3
  from scipy.io.wavfile import write as write_wav
4
  import tempfile
5
  import torch
 
6
 
7
  # Add NumPy scalar to safe globals to fix UnpicklingError
8
- torch.serialization.add_safe_globals(['numpy.core.multiarray.scalar'])
9
 
10
  # Preload the models at startup
11
  preload_models()
 
3
  from scipy.io.wavfile import write as write_wav
4
  import tempfile
5
  import torch
6
+ from numpy.core.multiarray import scalar
7
 
8
  # Add NumPy scalar to safe globals to fix UnpicklingError
9
+ torch.serialization.add_safe_globals([scalar])
10
 
11
  # Preload the models at startup
12
  preload_models()