Update app.py

app.py

@@ -7,6 +7,8 @@ from datasets import load_dataset
 from tqdm import tqdm
 import soundfile as sf
 import librosa
+import random
+
 
 dataset = load_dataset('SeyedAli/Persian-Speech-Dataset')
 
@@ -14,11 +16,15 @@ dataset = dataset["test"]
 
 
 def set_seed(seed):
-  torch.manual_seed(seed)
-  if torch.cuda.is_available():
-    torch.cuda.manual_seed_all(seed)
-
-set_seed(42)
+    torch.manual_seed(seed)
+    if torch.cuda.is_available():
+        torch.cuda.manual_seed_all(seed)
+    np.random.seed(seed)
+    random.seed(seed)
+    torch.backends.cudnn.deterministic = True
+    torch.backends.cudnn.benchmark = False
+
+set_seed(1)
 # Load model directly
 from transformers import AutoProcessor, AutoModelForTextToSpectrogram