Update app.py
Browse files
app.py
CHANGED
@@ -27,8 +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 |
-
#all-MiniLM-L6-v2, BAAI/bge-base-en-v1.5
|
32 |
#docs_text = [doc.text for doc in docs]
|
33 |
#embed = embedding_model.embed_documents(docs_text)
|
34 |
|
|
|
27 |
#docs = splitter.create_documents(str(dataset))
|
28 |
# Returns a list of documents
|
29 |
#print(docs)
|
30 |
+
embedding_model = SentenceTransformer("BAAI/bge-large-en-v1.5")
|
31 |
+
#all-MiniLM-L6-v2, BAAI/bge-base-en-v1.5,infgrad/stella-base-en-v2
|
32 |
#docs_text = [doc.text for doc in docs]
|
33 |
#embed = embedding_model.embed_documents(docs_text)
|
34 |
|