NickNYU commited on
Commit
9abd65a
1 Parent(s): 7607fec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -60,7 +60,10 @@ def main():
60
  if query:
61
  docs = VectorStore.similarity_search(query=query, k=3)
62
 
63
- llm = OpenAI()
 
 
 
64
  chain = load_qa_chain(llm=llm, chain_type="stuff")
65
  with get_openai_callback() as cb:
66
  response = chain.run(input_documents=docs, question=query)
 
60
  if query:
61
  docs = VectorStore.similarity_search(query=query, k=3)
62
 
63
+ llm = AzureOpenAI(
64
+ engine="SBUXOpenAI",
65
+ model_name="SBUXOpenAI",
66
+ )
67
  chain = load_qa_chain(llm=llm, chain_type="stuff")
68
  with get_openai_callback() as cb:
69
  response = chain.run(input_documents=docs, question=query)