Update app.py
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ def infer_birdcall(model, mp3_file, segment_length=500, device="cuda"):
|
|
58 |
sequence = torch.stack(segment_tensors, dim=0).unsqueeze(0).to(device)
|
59 |
output = model(sequence)
|
60 |
pred = torch.max(output, dim=1)[1].cpu().numpy()[0]
|
61 |
-
with open('
|
62 |
class_names = json.load(f)
|
63 |
predicted_bird = class_names[pred]
|
64 |
return predicted_bird
|
|
|
58 |
sequence = torch.stack(segment_tensors, dim=0).unsqueeze(0).to(device)
|
59 |
output = model(sequence)
|
60 |
pred = torch.max(output, dim=1)[1].cpu().numpy()[0]
|
61 |
+
with open('class_mapping.json', 'r') as f:
|
62 |
class_names = json.load(f)
|
63 |
predicted_bird = class_names[pred]
|
64 |
return predicted_bird
|