Commit
·
d4b788f
1
Parent(s):
1e8280a
Update document_questioner_app.py
Browse files
document_questioner_app.py
CHANGED
|
@@ -18,18 +18,15 @@ def question_document(Document, Question):
|
|
| 18 |
|
| 19 |
loader = PyPDFLoader(Document.name)
|
| 20 |
docs = loader.load()
|
| 21 |
-
|
| 22 |
-
# Set Open AI Key
|
| 23 |
-
os.environ["OPENAI_API_KEY"] = os.environ['OpenaiKey']
|
| 24 |
|
| 25 |
# Create embeddings
|
| 26 |
-
embeddings = OpenAIEmbeddings()
|
| 27 |
|
| 28 |
# Write in DB
|
| 29 |
docsearch = Chroma.from_documents(docs, embeddings, ids=["page" + str(d.metadata["page"]) for d in docs])
|
| 30 |
|
| 31 |
# Define LLM
|
| 32 |
-
llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.2)
|
| 33 |
|
| 34 |
# Customize map_reduce prompts
|
| 35 |
question_template = """{context}
|
|
|
|
| 18 |
|
| 19 |
loader = PyPDFLoader(Document.name)
|
| 20 |
docs = loader.load()
|
|
|
|
|
|
|
|
|
|
| 21 |
|
| 22 |
# Create embeddings
|
| 23 |
+
embeddings = OpenAIEmbeddings(openai_api_key = os.environ['OpenaiKey'])
|
| 24 |
|
| 25 |
# Write in DB
|
| 26 |
docsearch = Chroma.from_documents(docs, embeddings, ids=["page" + str(d.metadata["page"]) for d in docs])
|
| 27 |
|
| 28 |
# Define LLM
|
| 29 |
+
llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.2, openai_api_key = os.environ['OpenaiKey'])
|
| 30 |
|
| 31 |
# Customize map_reduce prompts
|
| 32 |
question_template = """{context}
|