Update asr.py

asr.py

@@ -1,5 +1,5 @@
 import librosa
-from transformers import pipeline, Wav2Vec2Processor, Wav2Vec2ForCTC
+from transformers import pipeline
 import logging
 
 # Set up logging
@@ -10,7 +10,7 @@ MODEL_ID = "facebook/mms-1b-all"
 
 try:
     # Create the pipeline with the appropriate model
-    pipe = pipeline("automatic-speech-recognition", model=MODEL_ID)
+    pipe = pipeline("automatic-speech-recognition", model=MODEL_ID, tokenizer=MODEL_ID)
     logging.info("ASR pipeline loaded successfully.")
 except Exception as e:
     logging.error(f"Error loading ASR pipeline: {e}")
@@ -35,7 +35,7 @@ def transcribe(audio):
         
         # Process the audio with the pipeline
         try:
-            transcription = pipe(audio_samples, return_tensors="pt", language="fao")["text"]
+            transcription = pipe(audio_samples, chunk_length_s=10, stride_length_s=5)["text"]
         except Exception as e:
             logging.error(f"Error during transcription with pipeline: {e}")
             return f"ERROR: Transcription failed - {e}"