MorettiGS commited on
Commit
77fe8b3
1 Parent(s): 19e3424

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,6 @@ from fastai.learner import load_learner
5
  import pandas as pd
6
  import numpy as np
7
 
8
- learn = load_learner('model.pkl')
9
-
10
  dados = pd.read_csv('valid.csv')
11
  ids = dados['user'].unique()
12
  ids_list = list(map(str, ids.tolist()))
@@ -43,6 +41,8 @@ class DotProduct(Module):
43
  res += self.user_bias(x[:,0]) + self.books_bias(x[:,1])
44
  return sigmoid_range(res, *self.y_range)
45
 
 
 
46
  def top5(user):
47
  user = int(user)
48
  items = pd.Series(learn.dls.classes['title']).unique()
 
5
  import pandas as pd
6
  import numpy as np
7
 
 
 
8
  dados = pd.read_csv('valid.csv')
9
  ids = dados['user'].unique()
10
  ids_list = list(map(str, ids.tolist()))
 
41
  res += self.user_bias(x[:,0]) + self.books_bias(x[:,1])
42
  return sigmoid_range(res, *self.y_range)
43
 
44
+ learn = load_learner('model.pkl', cpu=True)
45
+
46
  def top5(user):
47
  user = int(user)
48
  items = pd.Series(learn.dls.classes['title']).unique()