avi292423 commited on
Commit
30fbe5a
·
verified ·
1 Parent(s): cd49382

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ label_map = {
27
  index_to_label = {v: k for k, v in label_map.items()}
28
 
29
  num_classes = 31
30
- pretrained_model = "facebook/wav2vec2-base" # Use base for less RAM, or keep large if needed
31
  model = Wav2VecIntent(num_classes=num_classes, pretrained_model=pretrained_model).to(device)
32
  state_dict = torch.load(MODEL_PATH, map_location=device)
33
  model.load_state_dict(state_dict)
 
27
  index_to_label = {v: k for k, v in label_map.items()}
28
 
29
  num_classes = 31
30
+ pretrained_model = "facebook/wav2vec2-large" # Use base for less RAM, or keep large if needed
31
  model = Wav2VecIntent(num_classes=num_classes, pretrained_model=pretrained_model).to(device)
32
  state_dict = torch.load(MODEL_PATH, map_location=device)
33
  model.load_state_dict(state_dict)