Cocoonai commited on
Commit
0f46ea8
·
verified ·
1 Parent(s): b411729

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -1,10 +1,7 @@
1
 
2
  import gradio as gr
3
- from InstructorEmbedding import INSTRUCTOR
4
- from embedding_utils import load_documents, embed_documents, create_vector_db, query_db
5
-
6
- model = INSTRUCTOR("hkunlp/instructor-base")
7
-
8
  documents = load_documents("vault")
9
  texts, embeddings, metadatas = embed_documents(documents, model)
10
  collection = create_vector_db(texts, embeddings, metadatas)
 
1
 
2
  import gradio as gr
3
+ from sentence_transformers import SentenceTransformer
4
+ model = SentenceTransformer("all-MiniLM-L6-v2")
 
 
 
5
  documents = load_documents("vault")
6
  texts, embeddings, metadatas = embed_documents(documents, model)
7
  collection = create_vector_db(texts, embeddings, metadatas)