Hugging Face's logo

Spaces:
heaversm
/
Codebert-Repo-Analyzer
Runtime error

App Files Community
heaversm commited on
Commit
aa527f7
·
verified ·
1 Parent(s): 4f03891

address faiss load local error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def get_similar_links(query, db, embeddings):
44
 
45
 
46
  embedding_vector = get_hugging_face_model()
47
- db = FAISS.load_local("code_sim_index", embedding_vector)
48
 
49
  st.title("📒 DSASearch Engine 🤖 ")
50
  text_input = st.text_area("Enter a Code Example", value =
 
44
 
45
 
46
  embedding_vector = get_hugging_face_model()
47
+ db = FAISS.load_local("code_sim_index", embedding_vector, allow_dangerous_deserialization=True)
48
 
49
  st.title("📒 DSASearch Engine 🤖 ")
50
  text_input = st.text_area("Enter a Code Example", value =