Cheselle commited on
Commit
24d1bcf
·
verified ·
1 Parent(s): 0861511

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -41,8 +41,11 @@ 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
  vectorstore = Qdrant.from_documents(
48
  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
+ #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,