Update app.py

app.py

@@ -177,7 +177,7 @@ def demo():
 
         # document = list_file_obj
         vector_db, collection_name = initialize_database(list_file_obj)
-
+        print(len(vector_db))
         # #upload_btn.upload(upload_file, inputs=[upload_btn], outputs=[document])
         # db_btn.click(initialize_database, \
         #     inputs=[document], \
@@ -190,8 +190,15 @@ def demo():
         #     outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
         #     queue=False)    
 
-        qachain = initialize_LLM(vector_db)
-        
+        #qachain = initialize_LLM(vector_db)
+        llm = HuggingFaceHub(repo_id = llm_model,model_kwargs={"temperature": temperature, 
+                                                           "max_new_tokens": max_tokens, 
+                                                           "top_k": top_k, 
+                                                           "load_in_8bit": True})
+        retriever=vector_db.as_retriever()
+        memory = ConversationBufferMemory(memory_key="chat_history", output_key='answer', return_messages=True)
+        qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff", 
+                                                         memory=memory,return_source_documents=True,verbose=False,)
         # Chatbot events
         msg.submit(conversation, \
             inputs=[qa_chain, msg, chatbot], \