Cheselle commited on
Commit
fe32e45
·
verified ·
1 Parent(s): e2651ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -42,14 +42,14 @@ recursive_blueprint_document = metadata_generator(ai_blueprint_document, "AI Blu
42
  combined_documents = recursive_framework_document + recursive_blueprint_document
43
 
44
 
45
- from transformers import AutoTokenizer, AutoModel
46
- import torch
47
- embeddings = 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(ai_framework_document + ai_blueprint_document)
53
 
54
  vectorstore = Qdrant.from_documents(
55
  documents=combined_documents,
 
42
  combined_documents = recursive_framework_document + recursive_blueprint_document
43
 
44
 
45
+ #from transformers import AutoTokenizer, AutoModel
46
+ #import torch
47
+ #embeddings = 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(ai_framework_document + ai_blueprint_document)
53
 
54
  vectorstore = Qdrant.from_documents(
55
  documents=combined_documents,