Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
48 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
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,
|