spycoder commited on
Commit
547b1d3
·
1 Parent(s): 50f2862

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,9 +16,9 @@ from collections import Counter
16
  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_classifier10.pth'
22
 
23
  if os.path.exists(model_path):
24
  print(f"Loading saved model {model_path}")
 
16
  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
 
23
  if os.path.exists(model_path):
24
  print(f"Loading saved model {model_path}")