7H4M3R commited on
Commit
1fd173a
·
verified ·
1 Parent(s): 2dd9c4b

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +5 -4
src/streamlit_app.py CHANGED
@@ -138,16 +138,17 @@ if st.button("Analyze"):
138
 
139
  with st.spinner("Classifying accent..."):
140
 
 
 
141
  waves = f"{np.concatenate(audio_df["audio"][:5].to_list())}"
142
  st.markdown("**Audio waves:**")
143
  st.text_area("Audio waves", waves, height=200)
 
 
144
 
145
  model_name = "dima806/english_accents_classification"
146
- pipe = pipeline('audio-classification', model=model_name, device=0) # GPU (device=0) or CPU (device=-1)
147
  accent_data = accent_classify(pipe, audio_path)
148
- audio_df = split_audio(audio_path)
149
- # audio_df = split_audio(audio_path)
150
- # print(np.concatenate(audio_df["audio"][:50].to_list()))
151
 
152
  # accent_data = {"label": "American", "score": 0.9}
153
  accent = accent_data.get("label", "American")
 
138
 
139
  with st.spinner("Classifying accent..."):
140
 
141
+ audio_df = split_audio(audio_path)
142
+ # print(np.concatenate(audio_df["audio"][:50].to_list()))
143
  waves = f"{np.concatenate(audio_df["audio"][:5].to_list())}"
144
  st.markdown("**Audio waves:**")
145
  st.text_area("Audio waves", waves, height=200)
146
+
147
+
148
 
149
  model_name = "dima806/english_accents_classification"
150
+ pipe = pipeline('audio-classification', model=model_name, device=0)
151
  accent_data = accent_classify(pipe, audio_path)
 
 
 
152
 
153
  # accent_data = {"label": "American", "score": 0.9}
154
  accent = accent_data.get("label", "American")