Update app.py

app.py

@@ -109,7 +109,7 @@ if __name__ == "__main__":
 
     run_name = "wav2vec"
 
-    checkpoint_path = "./checkpoints/wav2vec-epoch=epoch=4.ckpt.ckpt"
+    checkpoint_path = "./wav2vec-epoch=epoch=4.ckpt.ckpt"
     checkpoint = torch.load(checkpoint_path)
     model.load_state_dict(checkpoint['state_dict'])
     trainer = Trainer(
@@ -118,12 +118,7 @@ if __name__ == "__main__":
 
     #trainer.fit(model, train_dataloader=trainloader, val_dataloaders=valloader)
     #trainer.test(model,dataloaders=testloader,verbose=True)
-    wav_path = "./speech-to-intent/wav_audios/92145547-3ab6-44e0-9245-085642fc4318.wav"
-    resmaple = torchaudio.transforms.Resample(8000, 16000)
-    wav_tensor,_ = torchaudio.load(wav_path)
-    wav_tensor = resmaple(wav_tensor)
-    model = model.to('cuda') 
-    y_hat = model.predict(wav_tensor)
+
     #with torch.no_grad():
      #   y_hat = model(wav_tensor)