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
|