Spaces:
Running
Running
Commit
·
673a9f6
1
Parent(s):
e5df0d0
Update app.py
Browse files
app.py
CHANGED
@@ -13,10 +13,10 @@ from langchain.chat_models import ChatOpenAI
|
|
13 |
|
14 |
# from langchain.chat_models import AzureChatOpenAI
|
15 |
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
openai.api_key = os.environ["OPENAI_API_KEY"]
|
20 |
|
21 |
|
22 |
def upload_pdf(file, pdf_text, embeddings, vectorstore, azure_embeddings, qa, progress = gr.Progress(track_tqdm=True)):
|
@@ -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 |
-
qa = RetrievalQA.from_chain_type(llm=
|
63 |
-
|
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 |
|
|
|
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 |
|
22 |
def upload_pdf(file, pdf_text, embeddings, vectorstore, azure_embeddings, qa, progress = gr.Progress(track_tqdm=True)):
|
|
|
59 |
documents=texts,
|
60 |
embeddings=embeddings,
|
61 |
metadatas=[{"source": "source"} for text in texts])
|
62 |
+
# qa = RetrievalQA.from_chain_type(llm= Chat /OpenAI(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 |
|