Namitg02 commited on
Commit
75679e8
·
verified ·
1 Parent(s): 4daef91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("infgrad/stella-base-en-v2")
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