Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,18 +5,18 @@ from sentence_transformers import SentenceTransformer, util
|
|
| 5 |
import pickle
|
| 6 |
|
| 7 |
# Load data
|
| 8 |
-
data = pd.read_csv("
|
| 9 |
titles = data["titles"]
|
| 10 |
|
| 11 |
# Load pre-trained SentenceTransformer model
|
| 12 |
model = SentenceTransformer("all-MiniLM-L6-v2")
|
| 13 |
|
| 14 |
# Load saved embeddings
|
| 15 |
-
with open("
|
| 16 |
Lencode = pickle.load(f)
|
| 17 |
|
| 18 |
# Load saved model
|
| 19 |
-
with open("
|
| 20 |
lModelRec = pickle.load(f)
|
| 21 |
|
| 22 |
def recomm(inputPaper):
|
|
|
|
| 5 |
import pickle
|
| 6 |
|
| 7 |
# Load data
|
| 8 |
+
data = pd.read_csv("./arxiv_data.csv")
|
| 9 |
titles = data["titles"]
|
| 10 |
|
| 11 |
# Load pre-trained SentenceTransformer model
|
| 12 |
model = SentenceTransformer("all-MiniLM-L6-v2")
|
| 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):
|