Update app.py

app.py

@@ -12,13 +12,14 @@ import torch.nn.functional as F
 from torch.utils.data import Dataset, DataLoader
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
 from collections import Counter
+
+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"
 if os.path.exists(model_path):
     print(f"Loading saved model {model_path}")
     model.load_state_dict(torch.load(model_path))
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 def predict(file_path):
     max_length = 100000