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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -47,9 +47,13 @@ combined_documents = recursive_framework_document + recursive_blueprint_document
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,
 
47
  #embeddings = AutoModel.from_pretrained("Cheselle/finetuned-arctic-sentence")
48
  #tokenizer = AutoTokenizer.from_pretrained("Cheselle/finetuned-arctic-sentence")
49
 
50
+ ai_framework_text = ai_framework_document.get_text("text")
51
+ ai_blueprint_text = ai_blueprint_document.get_text("text")
52
+
53
  from sentence_transformers import SentenceTransformer
54
  embedding_model = SentenceTransformer("Cheselle/finetuned-arctic-sentence")
55
+ embeddings = embedding_model.encode(ai_framework_text + ai_blueprint_text)
56
+ #embeddings = embedding_model.encode(ai_framework_document + ai_blueprint_document)
57
 
58
  vectorstore = Qdrant.from_documents(
59
  documents=combined_documents,