1998Shubham commited on
Commit
37551d3
·
verified ·
1 Parent(s): 3713abf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -15,12 +15,9 @@ model = SentenceTransformer("all-MiniLM-L6-v2")
15
  with open("/content/embedding.pkl", "rb") as f:
16
  Lencode = pickle.load(f)
17
 
18
- # Load saved model
19
- with open("/content/ModelRec.pkl", "rb") as f:
20
- lModelRec = pickle.load(f)
21
 
22
  def recomm(inputPaper):
23
- encodePaper = lModelRec.encode(inputPaper)
24
  cosine_score = util.cos_sim(Lencode, encodePaper)
25
  top_scores = torch.topk(cosine_score, dim=0, k=4)
26
  paperList = []
 
15
  with open("/content/embedding.pkl", "rb") as f:
16
  Lencode = pickle.load(f)
17
 
 
 
 
18
 
19
  def recomm(inputPaper):
20
+ encodePaper = model.encode(inputPaper)
21
  cosine_score = util.cos_sim(Lencode, encodePaper)
22
  top_scores = torch.topk(cosine_score, dim=0, k=4)
23
  paperList = []