Update app.py

app.py

@@ -4,7 +4,7 @@ import pickle
 import pandas as pd
 import torch
 
-model = torch.load('learn.pkl',map_location=torch.device('cpu') )
+learn = load_learner('exported_learner.pkl')
 
 def recommendation(user_id, book_1, book_2, book_3):
     data = {'user_id': [user_id, user_id, user_id], 
@@ -12,8 +12,8 @@ def recommendation(user_id, book_1, book_2, book_3):
 
     df = pd.DataFrame(data=data)
 
-    testdl = model.dls.test_dl(df)
-    preds = model.get_preds(dl=testdl)
+    testdl = learn.dls.test_dl(df)
+    preds = learn.get_preds(dl=testdl)
     preds
     
     rec = 0