Update app.py

app.py

@@ -2,10 +2,13 @@ from smolagents import CodeAgent, HfApiModel, load_tool, tool
 import datetime
 import pytz
 import yaml
-import whisper
+from transformers import pipeline
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+# Initialize ASR model from Hugging Face
+asr_pipeline = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-large-960h")
+
 # Tool: Convert time between time zones
 @tool
 def convert_time(time_str: str, from_tz: str, to_tz: str) -> str:
@@ -46,7 +49,7 @@ def assign_alert(alert_message: str) -> str:
     return "Alert could not be assigned automatically. Please check manually."
 
 
-# Tool: Transcribe audio and generate a timeline
+# Tool: Transcribe audio and generate a timeline using Hugging Face ASR
 @tool
 def transcribe_audio(audio_path: str) -> str:
     """Transcribe audio and log events with timestamps.
@@ -54,8 +57,7 @@ def transcribe_audio(audio_path: str) -> str:
         audio_path: Path to the audio file.
     """
     try:
-        model = whisper.load_model("base")
-        result = model.transcribe(audio_path)
+        result = asr_pipeline(audio_path)
         transcript = result["text"]
         timestamped_events = [f"{datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - {line}" for line in transcript.split('.')]
         return "\n".join(timestamped_events)