Update app.py

app.py

@@ -67,13 +67,19 @@ from io import BytesIO
 import librosa
 
 # Updated function for Streamlit-compatible audio processing
-def get_audio_embedding(uploaded_file):
-    # Convert uploaded file to in-memory buffer
-    audio_bytes = BytesIO(uploaded_file.read())
-    
-    # Load audio using librosa from the BytesIO buffer
-    audio, sr = librosa.load(audio_bytes, sr=16000)
-    
+import soundfile as sf
+from io import BytesIO
+
+# Update the function to handle both file paths and Streamlit-uploaded files
+def get_audio_embedding(file_input):
+    # Determine if input is a file path (string) or a file-like object (uploaded file)
+    if isinstance(file_input, str):
+        # If the input is a file path, use it directly with librosa
+        audio, sr = librosa.load(file_input, sr=16000)
+    else:
+        # If the input is an uploaded file (BytesIO), use soundfile to read it
+        audio, sr = sf.read(BytesIO(file_input.read()), dtype='float32')
+
     # Extract features using the Wav2Vec2 model
     inputs = feature_extractor(audio, sampling_rate=sr, return_tensors="pt", padding=True)
     with torch.no_grad():