prashant-garg commited on
Commit
622a5d0
·
1 Parent(s): 15d8269

broken indent fix

Browse files
Files changed (1) hide show
  1. app.py +19 -1
app.py CHANGED
@@ -5,7 +5,14 @@ import streamlit as st
5
  import numpy as np
6
  import torch
7
  from transformers import AutoFeatureExtractor, AutoModelForAudioClassification
 
 
 
 
 
8
  import pyaudio
 
 
9
  import logging
10
 
11
  # Configure logging
@@ -29,6 +36,15 @@ def load_model():
29
  model.eval()
30
  return feature_extractor, model
31
 
 
 
 
 
 
 
 
 
 
32
  placeholder = st.empty()
33
  placeholder.text("Loading model...")
34
  feature_extractor, model = load_model()
@@ -67,7 +83,9 @@ def start_listening():
67
  channels=CHANNELS,
68
  rate=RATE,
69
  input=True,
70
- frames_per_buffer=CHUNK)
 
 
71
 
72
  st.session_state['stream'] = stream
73
  st.session_state['audio'] = audio
 
5
  import numpy as np
6
  import torch
7
  from transformers import AutoFeatureExtractor, AutoModelForAudioClassification
8
+
9
+ import sys
10
+ import os
11
+ # Suppress ALSA warnings
12
+ sys.stderr = open(os.devnull, 'w')
13
  import pyaudio
14
+ sys.stderr = sys.__stderr__
15
+
16
  import logging
17
 
18
  # Configure logging
 
36
  model.eval()
37
  return feature_extractor, model
38
 
39
+ def see_audio_options():
40
+
41
+ p = pyaudio.PyAudio()
42
+ for i in range(p.get_device_count()):
43
+ info = p.get_device_info_by_index(i)
44
+ print(f"Device {i}: {info['name']}")
45
+
46
+ see_audio_options()
47
+
48
  placeholder = st.empty()
49
  placeholder.text("Loading model...")
50
  feature_extractor, model = load_model()
 
83
  channels=CHANNELS,
84
  rate=RATE,
85
  input=True,
86
+ frames_per_buffer=CHUNK,
87
+ device_index=0
88
+ )
89
 
90
  st.session_state['stream'] = stream
91
  st.session_state['audio'] = audio