Cheselle commited on
Commit
e797fa9
·
verified ·
1 Parent(s): 915e7fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -41,15 +41,15 @@ recursive_framework_document = metadata_generator(ai_framework_document, "AI Fra
41
  recursive_blueprint_document = metadata_generator(ai_blueprint_document, "AI Blueprint")
42
  combined_documents = recursive_framework_document + recursive_blueprint_document
43
 
44
- #from transformers import AutoModel
45
- #embeddings = AutoModel.from_pretrained("Cheselle/finetuned-arctic-sentence")
46
 
47
- from sentence_transformers import SentenceTransformer
48
- embeddings = SentenceTransformer("Cheselle/finetuned-arctic-sentence")
49
 
50
  vectorstore = Qdrant.from_documents(
51
  documents=combined_documents,
52
- embedding=embeddings.encode,
53
  location=":memory:",
54
  collection_name="ai_policy"
55
  )
 
41
  recursive_blueprint_document = metadata_generator(ai_blueprint_document, "AI Blueprint")
42
  combined_documents = recursive_framework_document + recursive_blueprint_document
43
 
44
+ from transformers import AutoModel
45
+ embeddings = AutoModel.from_pretrained("Cheselle/finetuned-arctic-sentence")
46
 
47
+ #from sentence_transformers import SentenceTransformer
48
+ #embeddings = SentenceTransformer("Cheselle/finetuned-arctic-sentence")
49
 
50
  vectorstore = Qdrant.from_documents(
51
  documents=combined_documents,
52
+ embedding=embeddings,
53
  location=":memory:",
54
  collection_name="ai_policy"
55
  )