Spaces:

VeenaGarre
/

Product_Recommendation

Runtime error

VeenaGarre commited on Dec 3, 2024

Commit

aec519c

verified ·

1 Parent(s): c55dca5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ if device.type == 'cuda':
     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

     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