Cheselle commited on
Commit
2b53a0b
·
verified ·
1 Parent(s): e797fa9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -41,11 +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,
 
41
  recursive_blueprint_document = metadata_generator(ai_blueprint_document, "AI Blueprint")
42
  combined_documents = recursive_framework_document + recursive_blueprint_document
43
 
 
 
44
 
45
+ #from transformers import AutoTokenizer, AutoModel
46
+ #import torch
47
+ #embedding = AutoModel.from_pretrained("Cheselle/finetuned-arctic-sentence")
48
+ #tokenizer = AutoTokenizer.from_pretrained("Cheselle/finetuned-arctic-sentence")
49
+
50
+ from sentence_transformers import SentenceTransformer
51
+ embedding_model = SentenceTransformer("Cheselle/finetuned-arctic-sentence")
52
+ embeddings = embedding_model.encode(combined_documents)
53
 
54
  vectorstore = Qdrant.from_documents(
55
  documents=combined_documents,