Update src/streamlit_app.py

src/streamlit_app.py

@@ -62,9 +62,9 @@ def split_audio(file):
         print(f"Error processing file: {e}")
         return None
 
-def accent_classify(pipe, audio_path):
-    audio_df = split_audio(audio_path)
-    return pipe(np.concatenate(audio_df["audio"][:150].to_list()))[0]
+# def accent_classify(pipe, audio_path):
+#     audio_df = split_audio(audio_path)
+#     return pipe(np.concatenate(audio_df["audio"][:250].to_list()))[0]
 
 accent_mapping = {
     'us': 'American',
@@ -94,11 +94,15 @@ if st.button("Analyze"):
 #             result = whisper_model.transcribe(audio_path)
 #             transcription = result['text']
 #             # pass
-
+        with st.spinner("Extracting waves..."):
+            audio_df = split_audio(audio_path)
+            waves = np.concatenate(audio_df["audio"][:250].to_list())
+            
         with st.spinner("Classifying accent..."):
             model_name = "dima806/english_accents_classification"
             pipe = pipeline('audio-classification', model=model_name, device=0)
-            accent_data = accent_classify(pipe, audio_path)
+            # accent_data = accent_classify(pipe, audio_path)
+            accent_data = pipe(waves)[0]
             accent = accent_mapping.get(accent_data.get("label", "us"))
             confidence = accent_data.get("score", 0)