cloud-sean commited on
Commit
673a9f6
·
1 Parent(s): e5df0d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -13,10 +13,10 @@ from langchain.chat_models import ChatOpenAI
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,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= 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
 
 
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