1998Shubham commited on
Commit
0912f5a
·
verified ·
1 Parent(s): 4695136

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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("/content/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("/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):
 
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):