bstraehle commited on
Commit
45e2e37
·
1 Parent(s): fff7464

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def invoke(openai_api_key, youtube_url, process_video, prompt):
37
  splits = text_splitter.split_documents(docs)
38
  else:
39
  print(222)
40
- splits = None
41
  vector_db = Chroma.from_documents(documents = splits, embedding = OpenAIEmbeddings(), persist_directory = CHROMA_DIR)
42
  llm = ChatOpenAI(model_name = MODEL_NAME, temperature = 0)
43
  qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
 
37
  splits = text_splitter.split_documents(docs)
38
  else:
39
  print(222)
40
+ splits = iter([])
41
  vector_db = Chroma.from_documents(documents = splits, embedding = OpenAIEmbeddings(), persist_directory = CHROMA_DIR)
42
  llm = ChatOpenAI(model_name = MODEL_NAME, temperature = 0)
43
  qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})