Update app.py

app.py

@@ -16,6 +16,9 @@ if device.type == 'cuda':
     print('Memory Usage:')
     print('Allocated:', round(torch.cuda.memory_allocated(0)/1024**3,1), 'GB')
     print('Cached:   ', round(torch.cuda.memory_reserved(0)/1024**3,1), 'GB')
+
+ncf_model.to(device)
+
 #Load dataset
 import pandas as pd
 from datasets import load_dataset