Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
160 |
-
else:
|
161 |
-
|
162 |
-
return
|
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):
|