espejelomar commited on
Commit
ca970dc
1 Parent(s): 0b511d0

Update backend/utils.py

Browse files
Files changed (1) hide show
  1. 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/descriptions_emb_10000_examples_github.pt",
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/codesearchnet_10000_python_examples_github.csv")
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