Rajagopal commited on
Commit
fa7774b
·
1 Parent(s): 8d15101

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -11,6 +11,8 @@ from langchain.llms import OpenAI
11
 
12
  from langchain.embeddings import OpenAIEmbeddings
13
 
 
 
14
 
15
  from langchain.vectorstores import Chroma
16
 
@@ -20,8 +22,16 @@ def loading_pdf():
20
  return "Loading..."
21
 
22
  def pdf_changes(pdf_doc, open_ai_key):
23
- if openai_key is not None:
24
  os.environ['OPENAI_API_KEY'] = open_ai_key
 
 
 
 
 
 
 
 
25
  loader = OnlinePDFLoader(pdf_doc.name)
26
  documents = loader.load()
27
  text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
@@ -31,7 +41,7 @@ def pdf_changes(pdf_doc, open_ai_key):
31
  retriever = db.as_retriever()
32
  global qa
33
  qa = ConversationalRetrievalChain.from_llm(
34
- llm=OpenAI(temperature=0.5),
35
  retriever=retriever,
36
  return_source_documents=False)
37
  return "Ready"
 
11
 
12
  from langchain.embeddings import OpenAIEmbeddings
13
 
14
+ from langchain.llms import AzureOpenAI
15
+ import openai
16
 
17
  from langchain.vectorstores import Chroma
18
 
 
22
  return "Loading..."
23
 
24
  def pdf_changes(pdf_doc, open_ai_key):
25
+ if True:
26
  os.environ['OPENAI_API_KEY'] = open_ai_key
27
+ os.environ['OPENAI_API_TYPE'] = 'azure'
28
+ # API version to use (Azure has several)
29
+ os.environ['OPENAI_API_VERSION'] = '0301'
30
+ # base URL for your Azure OpenAI resource
31
+ os.environ['OPENAI_API_BASE'] = 'https://nand3.openai.azure.com'
32
+
33
+ llm = AzureOpenAI(deployment_name="Nand", model_name="gpt-35-turbo")
34
+
35
  loader = OnlinePDFLoader(pdf_doc.name)
36
  documents = loader.load()
37
  text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
 
41
  retriever = db.as_retriever()
42
  global qa
43
  qa = ConversationalRetrievalChain.from_llm(
44
+ llm=AzureOpenAI(deployment_name="Nand", model_name="gpt-35-turbo"),
45
  retriever=retriever,
46
  return_source_documents=False)
47
  return "Ready"