Update app.py
Browse files
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 =
|
| 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 = []
|