Update backend/utils.py

backend/utils.py

@@ -1,7 +1,7 @@
 import pandas as pd
-from sentence_transformers import SentenceTransformer
 import streamlit as st
 import torch
+from sentence_transformers import SentenceTransformer
 
 
 @st.cache(allow_output_mutation=True)
@@ -11,11 +11,10 @@ def load_model(model_name):
     return model
 
 
-@st.cache(allow_output_mutation=True)
 def load_embeddings():
     # embedding pre-generated
     corpus_emb = torch.load(
-        "./embeddings/descriptions_emb_100000_examples.pt",
+        "./embeddings/descriptions_emb_10000_examples_github.pt",
         map_location=torch.device("cpu"),
     )
     return corpus_emb
@@ -24,5 +23,5 @@ def load_embeddings():
 @st.cache(allow_output_mutation=True)
 def load_texts():
     # texts database pre-generated
-    corpus_texts = pd.read_csv("./data/codesearchnet_100000_examples.csv")
+    corpus_texts = pd.read_csv("./data/codesearchnet_10000_python_examples_github.csv")
     return corpus_texts