fevot commited on
Commit
23dc9ad
·
verified ·
1 Parent(s): 7d4aad7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('class_names.json', 'r') as f:
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