Ar4ikov commited on
Commit
55c2d79
1 Parent(s): feb9c03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -45,12 +45,12 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
45
  print(device)
46
 
47
 
48
- def recognize(audio):
49
  sr, audio_array = audio
50
  audio_array = audio_array.astype(np.float32)
51
  print(sr, audio_array)
52
  state = predict(audio_array, sr)
53
- return state
54
 
55
 
56
  def test_some(audio):
@@ -61,13 +61,15 @@ def test_some(audio):
61
 
62
 
63
  interface = gr.Interface(
64
- fn=test_some,
65
  inputs=[
66
- "microphone"
 
67
  ],
68
  outputs=[
69
- "audio"
 
70
  ],
71
- live=False)
72
 
73
  interface.launch(debug=True)
 
45
  print(device)
46
 
47
 
48
+ def recognize(audio, state={}):
49
  sr, audio_array = audio
50
  audio_array = audio_array.astype(np.float32)
51
  print(sr, audio_array)
52
  state = predict(audio_array, sr)
53
+ return state, state
54
 
55
 
56
  def test_some(audio):
 
61
 
62
 
63
  interface = gr.Interface(
64
+ fn=recognize,
65
  inputs=[
66
+ gr.Audio(source="microphone", label="Скажите что-нибудь..."),
67
+ "state"
68
  ],
69
  outputs=[
70
+ gr.Label(num_top_classes=7),
71
+ "state"
72
  ],
73
+ live=True)
74
 
75
  interface.launch(debug=True)