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):
|