HypermindLabs commited on
Commit
47906a4
·
1 Parent(s): d1170f5

big update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,11 +31,11 @@ def process_data(waveform_chunks):
31
  st.write("Reached stage 4")
32
  for chunk in waveform_chunks:
33
  st.write("Reached stage 5")
34
- input_tensor = torch.tensor(chunk).unsqueeze(0).to(torch.float32)
35
  st.write("Reached stage 6")
36
  st.write(input_tensor.shape)
37
  result = model(input_tensor)
38
- st.write("Reached stage 7")
39
  if np.abs(result[0][0]) > np.abs(result[0][1]):
40
  other += 1
41
  else:
 
31
  st.write("Reached stage 4")
32
  for chunk in waveform_chunks:
33
  st.write("Reached stage 5")
34
+ input_tensor = torch.tensor(chunk).unsqueeze(0)
35
  st.write("Reached stage 6")
36
  st.write(input_tensor.shape)
37
  result = model(input_tensor)
38
+ st.write(result)
39
  if np.abs(result[0][0]) > np.abs(result[0][1]):
40
  other += 1
41
  else: