Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,8 @@ dataset = load_dataset("not-lain/wikipedia",revision = "embedded")
|
|
| 27 |
#docs = splitter.create_documents(str(dataset))
|
| 28 |
# Returns a list of documents
|
| 29 |
#print(docs)
|
| 30 |
-
embedding_model = SentenceTransformer("
|
|
|
|
| 31 |
#docs_text = [doc.text for doc in docs]
|
| 32 |
#embed = embedding_model.embed_documents(docs_text)
|
| 33 |
|
|
|
|
| 27 |
#docs = splitter.create_documents(str(dataset))
|
| 28 |
# Returns a list of documents
|
| 29 |
#print(docs)
|
| 30 |
+
embedding_model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
|
| 31 |
+
#all-MiniLM-L6-v2
|
| 32 |
#docs_text = [doc.text for doc in docs]
|
| 33 |
#embed = embedding_model.embed_documents(docs_text)
|
| 34 |
|