Spaces:
Running
Running
Commit
·
62c0fcf
1
Parent(s):
48879d4
Update app.py
Browse files
app.py
CHANGED
@@ -7,15 +7,15 @@ from PyPDF2 import PdfReader
|
|
7 |
from langchain.text_splitter import CharacterTextSplitter, RecursiveCharacterTextSplitter
|
8 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
9 |
from langchain import VectorDBQA
|
10 |
-
from langchain.llms import
|
11 |
from langchain.chains import RetrievalQA
|
12 |
from langchain.chat_models import AzureChatOpenAI
|
13 |
|
14 |
# from langchain.chat_models import AzureChatOpenAI
|
15 |
|
16 |
-
os.environ["OPENAI_API_TYPE"] = openai.api_type = "azure"
|
17 |
-
os.environ["OPENAI_API_VERSION"] = openai.api_version = "2023-03-15-preview"
|
18 |
-
os.environ["OPENAI_API_BASE"] = openai.api_base = "https://cog-mnjbf5r4o6b3e.openai.azure.com/"
|
19 |
openai.api_key = os.environ["OPENAI_API_KEY"]
|
20 |
|
21 |
|
@@ -59,8 +59,8 @@ def upload_pdf(file, pdf_text, embeddings, vectorstore, azure_embeddings, qa, pr
|
|
59 |
documents=texts,
|
60 |
embeddings=embeddings,
|
61 |
metadatas=[{"source": "source"} for text in texts])
|
62 |
-
|
63 |
-
qa = RetrievalQA.from_chain_type(llm= AzureChatOpenAI(deployment_name="chat", model_name="gpt-35-turbo"), chain_type="stuff", vectorstore=vectorstore.as_retriever())
|
64 |
|
65 |
return pdf_text, pdf_text, embeddings, vectorstore, azure_embeddings, qa, gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
|
66 |
|
|
|
7 |
from langchain.text_splitter import CharacterTextSplitter, RecursiveCharacterTextSplitter
|
8 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
9 |
from langchain import VectorDBQA
|
10 |
+
from langchain.llms import ChatOpenAI
|
11 |
from langchain.chains import RetrievalQA
|
12 |
from langchain.chat_models import AzureChatOpenAI
|
13 |
|
14 |
# from langchain.chat_models import AzureChatOpenAI
|
15 |
|
16 |
+
# os.environ["OPENAI_API_TYPE"] = openai.api_type = "azure"
|
17 |
+
# os.environ["OPENAI_API_VERSION"] = openai.api_version = "2023-03-15-preview"
|
18 |
+
# os.environ["OPENAI_API_BASE"] = openai.api_base = "https://cog-mnjbf5r4o6b3e.openai.azure.com/"
|
19 |
openai.api_key = os.environ["OPENAI_API_KEY"]
|
20 |
|
21 |
|
|
|
59 |
documents=texts,
|
60 |
embeddings=embeddings,
|
61 |
metadatas=[{"source": "source"} for text in texts])
|
62 |
+
qa = RetrievalQA.from_chain_type(llm= ChatOpenAI(model_name='gpt-3.5-turbo' ), chain_type="stuff", retriever=vectorstore.as_retriever())
|
63 |
+
# qa = RetrievalQA.from_chain_type(llm= AzureChatOpenAI(deployment_name="chat", model_name="gpt-35-turbo"), chain_type="stuff", vectorstore=vectorstore.as_retriever())
|
64 |
|
65 |
return pdf_text, pdf_text, embeddings, vectorstore, azure_embeddings, qa, gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
|
66 |
|