1998Shubham commited on
Commit
bf81c5e
·
verified ·
1 Parent(s): 510ecf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,11 +13,11 @@ model = SentenceTransformer("all-MiniLM-L6-v2")
13
 
14
  # Load saved embeddings
15
  with open("./embedding.pkl", "rb") as f:
16
- Lencode = pickle.load(f, map_location=torch.device('cpu'))
17
 
18
  # Load saved model
19
  with open("./ModelRec.pkl", "rb") as f:
20
- lModelRec = pickle.load(f, map_location=torch.device('cpu'))
21
 
22
  def recomm(inputPaper):
23
  encodePaper = lModelRec.encode(inputPaper)
 
13
 
14
  # Load saved embeddings
15
  with open("./embedding.pkl", "rb") as f:
16
+ Lencode = pickle.load(f)
17
 
18
  # Load saved model
19
  with open("./ModelRec.pkl", "rb") as f:
20
+ lModelRec = pickle.load(f)
21
 
22
  def recomm(inputPaper):
23
  encodePaper = lModelRec.encode(inputPaper)