espejelomar
commited on
Commit
•
ca970dc
1
Parent(s):
0b511d0
Update backend/utils.py
Browse files- backend/utils.py +2 -2
backend/utils.py
CHANGED
@@ -14,7 +14,7 @@ def load_model(model_name):
|
|
14 |
def load_embeddings():
|
15 |
# embedding pre-generated
|
16 |
corpus_emb = torch.load(
|
17 |
-
"./embeddings/
|
18 |
map_location=torch.device("cpu"),
|
19 |
)
|
20 |
return corpus_emb
|
@@ -23,5 +23,5 @@ def load_embeddings():
|
|
23 |
@st.cache(allow_output_mutation=True)
|
24 |
def load_texts():
|
25 |
# texts database pre-generated
|
26 |
-
corpus_texts = pd.read_csv("./data/
|
27 |
return corpus_texts
|
|
|
14 |
def load_embeddings():
|
15 |
# embedding pre-generated
|
16 |
corpus_emb = torch.load(
|
17 |
+
"./embeddings/descriptions_emb_200000_examples_github.pt",
|
18 |
map_location=torch.device("cpu"),
|
19 |
)
|
20 |
return corpus_emb
|
|
|
23 |
@st.cache(allow_output_mutation=True)
|
24 |
def load_texts():
|
25 |
# texts database pre-generated
|
26 |
+
corpus_texts = pd.read_csv("./data/codesearchnet_200000_python_examples_github.csv")
|
27 |
return corpus_texts
|