demoPOC commited on
Commit
c9cbae2
·
1 Parent(s): 1c922c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -24,7 +24,7 @@ import openai
24
  from langchain.llms import OpenAI
25
 
26
  from langchain.embeddings.openai import OpenAIEmbeddings
27
- from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
28
 
29
  from langchain.document_loaders import SeleniumURLLoader, PyPDFLoader
30
 
@@ -155,11 +155,11 @@ def pretty_print_docs(docs):
155
 
156
 
157
  def getEmbeddingModel(embeddingId):
158
- if (embeddingId == 1):
159
- embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
160
- else:
161
- embeddings = OpenAIEmbeddings()
162
- return embeddings
163
 
164
 
165
  def getLLMModel(LLMID):
 
24
  from langchain.llms import OpenAI
25
 
26
  from langchain.embeddings.openai import OpenAIEmbeddings
27
+ #from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
28
 
29
  from langchain.document_loaders import SeleniumURLLoader, PyPDFLoader
30
 
 
155
 
156
 
157
  def getEmbeddingModel(embeddingId):
158
+ # if (embeddingId == 1):
159
+ # embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
160
+ # else:
161
+ # embeddings = OpenAIEmbeddings()
162
+ return OpenAIEmbeddings()
163
 
164
 
165
  def getLLMModel(LLMID):