demoPOC commited on
Commit
ce81c31
·
1 Parent(s): 35e83e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -183,19 +183,19 @@ def file_Upload():
183
 
184
  print("All chunk List END ***********************\n\n")
185
 
186
- #embeddings = OpenAIEmbeddings()
187
  #from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
188
  #embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
189
- from langchain.embeddings import HuggingFaceEmbeddings
190
-
191
- model_name = "sentence-transformers/all-MiniLM-L6-v2"
192
- model_kwargs = {'device': 'cpu'}
193
- encode_kwargs = {'normalize_embeddings': False}
194
- embeddings = HuggingFaceEmbeddings(
195
- model_name=model_name,
196
- model_kwargs=model_kwargs,
197
- encode_kwargs=encode_kwargs
198
- )
199
  global vectordb
200
  #vectordb = Chroma.from_documents(texts,embeddings)
201
  vectordb=Chroma.from_documents(documents=texts, embedding=embeddings, collection_metadata={"hnsw:space": "cosine"})
 
183
 
184
  print("All chunk List END ***********************\n\n")
185
 
186
+ embeddings = OpenAIEmbeddings()
187
  #from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
188
  #embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
189
+ # from langchain.embeddings import HuggingFaceEmbeddings
190
+
191
+ # model_name = "sentence-transformers/all-MiniLM-L6-v2"
192
+ # model_kwargs = {'device': 'cpu'}
193
+ # encode_kwargs = {'normalize_embeddings': False}
194
+ # embeddings = HuggingFaceEmbeddings(
195
+ # model_name=model_name,
196
+ # model_kwargs=model_kwargs,
197
+ # encode_kwargs=encode_kwargs
198
+ # )
199
  global vectordb
200
  #vectordb = Chroma.from_documents(texts,embeddings)
201
  vectordb=Chroma.from_documents(documents=texts, embedding=embeddings, collection_metadata={"hnsw:space": "cosine"})