Fixed bug in preprocessing script

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+.env

app.py

@@ -5,6 +5,13 @@ import numpy as np
 import webrtcvad
 from transformers import WhisperProcessor, WhisperForConditionalGeneration, pipeline
 
+from dotenv import load_dotenv
+import os
+
+load_dotenv()  # Loads the .env file
+token = os.getenv("HUGGINGFACE_TOKEN")
+
+
 # Model names
 TN_MODEL_NAME = "amenIKh/Tunisian_Checkpoint12"
 WHISPER_MODEL_NAME = "openai/whisper-small"
@@ -14,11 +21,13 @@ pipe_tn = pipeline(
     task="automatic-speech-recognition",
     model=TN_MODEL_NAME,
     device=0 if torch.cuda.is_available() else -1,
+    use_auth_token=token
+
 )
 
 # Load Whisper model and processor
-whisper_model = WhisperForConditionalGeneration.from_pretrained(WHISPER_MODEL_NAME)
-whisper_processor = WhisperProcessor.from_pretrained(WHISPER_MODEL_NAME)
+whisper_model = WhisperForConditionalGeneration.from_pretrained(WHISPER_MODEL_NAME,use_auth_token=token)
+whisper_processor = WhisperProcessor.from_pretrained(WHISPER_MODEL_NAME,use_auth_token=token)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 whisper_model.to(device)