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