spycoder commited on
Commit
2cbb9da
·
1 Parent(s): d9ba6ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ device = torch.device("cpu")
17
  processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
18
  model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/wav2vec2-base-960h", num_labels=2).to(device)
19
  # model_path = "dysarthria_classifier12.pth"
20
- model_path = 'model_weights2.pth'
21
- # model_path = '/home/user/app/dysarthria_classifier12.pth'
22
  model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
23
 
24
  # if os.path.exists(model_path):
 
17
  processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
18
  model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/wav2vec2-base-960h", num_labels=2).to(device)
19
  # model_path = "dysarthria_classifier12.pth"
20
+ # model_path = 'model_weights2.pth'
21
+ model_path = '/home/user/app/dysarthria_classifier10.pth'
22
  model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
23
 
24
  # if os.path.exists(model_path):