monet-joe commited on
Commit
0823851
β€’
1 Parent(s): 4a0d5fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -147,7 +147,7 @@ def inference(wav_path: str, log_name: str, folder_path="./tmp"):
147
  input = embed_img(file_path)
148
  output = model(input)
149
  pred_id = torch.max(output.data, 1)[1]
150
- outputs.append(pred_id)
151
 
152
  max_count_item = most_common_element(outputs)
153
  shutil.rmtree(folder_path)
 
147
  input = embed_img(file_path)
148
  output = model(input)
149
  pred_id = torch.max(output.data, 1)[1]
150
+ outputs.append(int(pred_id))
151
 
152
  max_count_item = most_common_element(outputs)
153
  shutil.rmtree(folder_path)