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