Update app.py
Browse files
app.py
CHANGED
@@ -42,16 +42,16 @@ import librosa
|
|
42 |
|
43 |
|
44 |
|
45 |
-
def set_seed(seed):
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
set_seed(42)
|
55 |
# Load model directly
|
56 |
from transformers import AutoProcessor, AutoModelForTextToSpectrogram
|
57 |
|
|
|
42 |
|
43 |
|
44 |
|
45 |
+
# def set_seed(seed):
|
46 |
+
# torch.manual_seed(seed)
|
47 |
+
# if torch.cuda.is_available():
|
48 |
+
# torch.cuda.manual_seed_all(seed)
|
49 |
+
# np.random.seed(seed)
|
50 |
+
# random.seed(seed)
|
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 |
|