monet-joe commited on
Commit
c234ee1
1 Parent(s): b4d0177

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -140,7 +140,7 @@ def inference(wav_path, log_name: str, folder_path="./tmp"):
140
  input = embed_img(file_path)
141
  output = model(input)
142
  pred_id = torch.max(output.data, 1)[1]
143
- outputs.append(pred_id)
144
 
145
  max_count_item = most_common_element(outputs)
146
  shutil.rmtree(folder_path)
 
140
  input = embed_img(file_path)
141
  output = model(input)
142
  pred_id = torch.max(output.data, 1)[1]
143
+ outputs.append(int(pred_id))
144
 
145
  max_count_item = most_common_element(outputs)
146
  shutil.rmtree(folder_path)