Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 from faster_whisper import WhisperModel
 import tempfile
 import logging
+import os
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -13,18 +14,18 @@ MODELS = {
     "Faster Whisper Medium": "Systran/faster-whisper-medium",  # Use the medium model
 }
 
-def transcribe_live_audio(audio, model_size="Faster Whisper Medium"):
+def transcribe_live_audio(audio):
     """Transcribe live audio from the microphone."""
     try:
         # Save the live audio to a temporary file
         with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_audio:
             temp_audio_path = temp_audio.name
-            audio.export(temp_audio_path, format="wav")
+            audio.save(temp_audio_path)  # Save the audio file
         
         # Load the appropriate model
         device = "cuda" if torch.cuda.is_available() else "cpu"
         compute_type = "float32" if device == "cuda" else "int8"
-        model = WhisperModel(MODELS[model_size], device=device, compute_type=compute_type)
+        model = WhisperModel(MODELS["Faster Whisper Medium"], device=device, compute_type=compute_type)
         
         # Transcribe the live audio
         segments, info = model.transcribe(