Update main.py

main.py

@@ -11,8 +11,15 @@ import io
 import random
 import tempfile
 import os
+import numba
+import soundfile as sf 
+
+os.environ['NUMBA_DISABLE_INTEL_SVML'] = '1'
+os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
+os.environ['NUMBA_CACHE_DIR'] = '/tmp/numba_cache'
+
+numba.config.DISABLE_JIT = True
 
-os.environ['NUMBA_DISABLE_JIT'] = '1'
 MultiPartParser.max_file_size = 200 * 1024 * 1024
 
 # Initialize FastAPI app
@@ -101,8 +108,11 @@ async def upload_audio(file: UploadFile = File(...)):
     print("Uploaded file saved to:", temp_file_path)
     
     # Load the audio file using librosa with a fixed sample rate of 16000 Hz
-    audio, sr = librosa.load(temp_file_path, sr=16000)
-        
+    audio, sr = sf.read(temp_file_path)
+    if sr != 16000:
+    # Resample if necessary
+        audio = librosa.resample(audio, sr, 16000)
+        sr = 16000
         # Process the audio data as needed
         # For this example, we'll just return some basic information
     duration = librosa.get_duration(y=audio, sr=sr)