Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +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(search_kwargs = {"k": 3}), 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>
|
|
|
44 |
llm = ChatOpenAI(model_name = MODEL_NAME, temperature = 0)
|
45 |
qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(search_kwargs = {"k": 3}), 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>
|