Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def set_seed(seed):
|
|
51 |
torch.backends.cudnn.deterministic = True
|
52 |
torch.backends.cudnn.benchmark = False
|
53 |
|
54 |
-
set_seed(
|
55 |
# Load model directly
|
56 |
from transformers import AutoProcessor, AutoModelForTextToSpectrogram
|
57 |
|
|
|
51 |
torch.backends.cudnn.deterministic = True
|
52 |
torch.backends.cudnn.benchmark = False
|
53 |
|
54 |
+
set_seed(42)
|
55 |
# Load model directly
|
56 |
from transformers import AutoProcessor, AutoModelForTextToSpectrogram
|
57 |
|