Update app.py

app.py

@@ -103,7 +103,6 @@ class LightningModel(pl.LightningModule):
         return predicted_class
 
 
-print(torch.cuda.mem_get_info())
 
 model = LightningModel()
 
@@ -112,9 +111,7 @@ run_name = "wav2vec"
 checkpoint_path = "./wav2vec-epoch=epoch=4.ckpt.ckpt"
 checkpoint = torch.load(checkpoint_path)
 model.load_state_dict(checkpoint['state_dict'])
-trainer = Trainer(
-    gpus=1
-)
+trainer = Trainer()
 
 #trainer.fit(model, train_dataloader=trainloader, val_dataloaders=valloader)
 #trainer.test(model,dataloaders=testloader,verbose=True)