Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ def set_seed(seed):
|
|
46 |
if torch.cuda.is_available():
|
47 |
torch.cuda.manual_seed_all(seed)
|
48 |
|
49 |
-
set_seed(
|
50 |
# Load model directly
|
51 |
from transformers import AutoProcessor, AutoModelForTextToSpectrogram
|
52 |
|
|
|
46 |
if torch.cuda.is_available():
|
47 |
torch.cuda.manual_seed_all(seed)
|
48 |
|
49 |
+
set_seed(42)
|
50 |
# Load model directly
|
51 |
from transformers import AutoProcessor, AutoModelForTextToSpectrogram
|
52 |
|