NHZ commited on
Commit
b891ccb
·
verified ·
1 Parent(s): f8ec047

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -73,11 +73,10 @@ def create_vector_store(text):
73
  embeddings = model(**tokens).last_hidden_state.mean(dim=1).squeeze().numpy()
74
  return embeddings
75
 
76
- # Create a function that directly returns embeddings
77
- embedding_function = lambda x: embed(x)
78
-
79
  # Create a FAISS vector store
80
- vector_store = FAISS.from_texts(texts=sentences, embedding=embedding_function)
 
 
81
 
82
  return vector_store, sentences
83
 
 
73
  embeddings = model(**tokens).last_hidden_state.mean(dim=1).squeeze().numpy()
74
  return embeddings
75
 
 
 
 
76
  # Create a FAISS vector store
77
+ vector_store = FAISS.from_texts(
78
+ texts=sentences, embedding=lambda x: embed(x)
79
+ )
80
 
81
  return vector_store, sentences
82