Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ def set_seed(seed):
|
|
49 |
np.random.seed(seed)
|
50 |
torch.backends.cudnn.benchmark = False
|
51 |
|
52 |
-
set_seed(
|
53 |
# Load model directly
|
54 |
from transformers import AutoProcessor, AutoModelForTextToSpectrogram
|
55 |
|
|
|
49 |
np.random.seed(seed)
|
50 |
torch.backends.cudnn.benchmark = False
|
51 |
|
52 |
+
set_seed(1)
|
53 |
# Load model directly
|
54 |
from transformers import AutoProcessor, AutoModelForTextToSpectrogram
|
55 |
|