Update stt.py

stt.py

@@ -1,6 +1,7 @@
-import tempfilehttps://huggingface.co/spaces/gk2410/AIMLInterviewer/settings
+import tempfile
 from faster_whisper import WhisperModel
 
+# Load model once (CPU-safe for Hugging Face Spaces)
 model = WhisperModel(
     "base",
     device="cpu",
@@ -8,11 +9,16 @@ model = WhisperModel(
 )
 
 def transcribe_audio(uploaded_file):
-    # Save Streamlit UploadedFile to temp file
+    """
+    Takes a Streamlit UploadedFile and returns transcribed text
+    """
+
+    # Save uploaded audio to a temp file
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
         tmp.write(uploaded_file.read())
         audio_path = tmp.name
 
     segments, _ = model.transcribe(audio_path)
 
-    return " ".join(seg.text for seg in segments).strip()
+    text = " ".join(segment.text for segment in segments)
+    return text.strip()