Spaces:
Running
Running
Monet Joe
commited on
Commit
•
0823851
1
Parent(s):
4a0d5fd
Update app.py
Browse files
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)
|