Update app.py
Browse files
app.py
CHANGED
|
@@ -44,6 +44,7 @@ def invoke(openai_api_key, youtube_url, process_video, prompt):
|
|
| 44 |
llm = ChatOpenAI(model_name = MODEL_NAME, temperature = 0)
|
| 45 |
qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
|
| 46 |
result = qa_chain({"query": prompt})
|
|
|
|
| 47 |
return result["result"]
|
| 48 |
|
| 49 |
description = """<strong>Overview:</strong> The app demonstrates how to use a <strong>Large Language Model</strong> (LLM) with <strong>Retrieval Augmented Generation</strong>
|
|
|
|
| 44 |
llm = ChatOpenAI(model_name = MODEL_NAME, temperature = 0)
|
| 45 |
qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
|
| 46 |
result = qa_chain({"query": prompt})
|
| 47 |
+
print(result)
|
| 48 |
return result["result"]
|
| 49 |
|
| 50 |
description = """<strong>Overview:</strong> The app demonstrates how to use a <strong>Large Language Model</strong> (LLM) with <strong>Retrieval Augmented Generation</strong>
|