Update stt.py

stt.py

@@ -1,7 +1,7 @@
 import os
 import torch
 import torchaudio
-import spaces  # Import spaces module for Zero-GPU
+import spaces
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
 
 # Create directories
@@ -20,8 +20,9 @@ WHISPER_MODEL_SIZES = {
     'large': 'openai/whisper-large-v3',
 }
 
-@spaces.GPU  # Add spaces.GPU decorator for Zero-GPU support
-async def transcribe_audio(audio_file_path, model_size="base", language="en"):
+# Synchronous function with GPU decorator
+@spaces.GPU
+def _transcribe_audio_gpu(audio_file_path, model_size="base", language="en"):
     global whisper_model, whisper_processor
     
     try:
@@ -73,4 +74,9 @@ async def transcribe_audio(audio_file_path, model_size="base", language="en"):
         
     except Exception as e:
         print(f"Error during transcription: {str(e)}")
-        return ""
+        return ""
+
+# Async wrapper that calls the GPU function
+async def transcribe_audio(audio_file_path, model_size="base", language="en"):
+    # Call the GPU-decorated function
+    return _transcribe_audio_gpu(audio_file_path, model_size, language)