shubhamrazzsharma commited on
Commit
978a35c
Β·
verified Β·
1 Parent(s): beecf40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ class GenreCNN(nn.Module):
61
  # ── Load model once at startup ────────────────────────────────────────────────
62
  model = GenreCNN().to(DEVICE)
63
  model.load_state_dict(
64
- torch.load("best_model (1).pth", map_location=DEVICE)
65
  )
66
  model.eval()
67
  print("βœ“ Model loaded successfully")
 
61
  # ── Load model once at startup ────────────────────────────────────────────────
62
  model = GenreCNN().to(DEVICE)
63
  model.load_state_dict(
64
+ torch.load("best_model.pth", map_location=DEVICE)
65
  )
66
  model.eval()
67
  print("βœ“ Model loaded successfully")