Update app.py

app.py

@@ -7,6 +7,7 @@ from datasets import load_dataset
 from tqdm import tqdm
 import soundfile as sf
 import librosa
+import random
 
 dataset = load_dataset('pourmand1376/asr-farsi-youtube-chunked-10-seconds', split = "test")
 
@@ -42,11 +43,15 @@ import librosa
 
 
 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