Update app.py

app.py

@@ -17,6 +17,9 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
 model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/wav2vec2-base-960h", num_labels=2).to(device)
 model_path = "dysarthria_classifier12.pth"
+model_path = '/home/user/app/dysarthria_classifier12.pth'
+model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
+
 if os.path.exists(model_path):
     print(f"Loading saved model {model_path}")
     model.load_state_dict(torch.load(model_path))