Update app.py

app.py

@@ -45,7 +45,7 @@ def invoke(openai_api_key, use_rag, prompt):
         if (os.path.isdir(CHROMA_DIR)):
             vector_db = Chroma(embedding_function = OpenAIEmbeddings(),
                                persist_directory = CHROMA_DIR)
-            print(os.listdir("Load DB"))
+            print("Load DB")
         else:
             loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
                                    OpenAIWhisperParser())
@@ -56,7 +56,7 @@ def invoke(openai_api_key, use_rag, prompt):
             vector_db = Chroma.from_documents(documents = splits, 
                                               embedding = OpenAIEmbeddings(), 
                                               persist_directory = CHROMA_DIR)
-            print(os.listdir("Make DB"))
+            print("Make DB")
         rag_chain = RetrievalQA.from_chain_type(llm, 
                                                 chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT}, 
                                                 retriever = vector_db.as_retriever(search_kwargs = {"k": 3}),