Spaces:
Running
Running
Update app.py
Browse files
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)
|