app.py
CHANGED
@@ -18,7 +18,7 @@ def classify_sound(file_path):
|
|
18 |
wv = wv.mean(dim=0, keepdim=True)
|
19 |
|
20 |
inputs = feature_extractor(
|
21 |
-
wv.squeeze().numpy(), sampling_rate=
|
22 |
)
|
23 |
|
24 |
with torch.no_grad():
|
|
|
18 |
wv = wv.mean(dim=0, keepdim=True)
|
19 |
|
20 |
inputs = feature_extractor(
|
21 |
+
wv.squeeze().numpy(), sampling_rate=16000, return_tensors="pt"
|
22 |
)
|
23 |
|
24 |
with torch.no_grad():
|