Update app.py

app.py

@@ -50,11 +50,10 @@ PROMPT = PromptTemplate(
 )
 chain_type_kwargs = {"prompt": PROMPT}
 
-llm = HuggingFaceHub(repo_id="mistralai/Mistral-7B-v0.1", model_kwargs={"temperature":0.1, "max_new_tokens":250})
+llm = HuggingFaceHub(repo_id="NousResearch/Llama-2-13b-hf", model_kwargs={"temperature":0.1, "max_new_tokens":250})
 
-# qachain=RetrievalQA.from_chain_type(llm, retriever=vectorstore.as_retriever(), memory=memory, chain_type_kwargs=chain_type_kwargs)
+qachain=RetrievalQA.from_chain_type(llm, retriever=vectorstore.as_retriever(), memory=memory, chain_type_kwargs=chain_type_kwargs)
 
-llm_chain = LLMChain(prompt=PROMPT , llm=llm)
 
 st.header("#CodeWars localGPT", divider='rainbow')
 
@@ -67,6 +66,6 @@ context = [] # the context stores a conversation history, you can use this to ma
 if(prompt):
     with st.chat_message(option):
          st.write(f"{datetime.datetime.now()} :red[{option}:] ", prompt)
-    context = llm_chain.run(prompt)
+    context = qachain({"query": user_input})
     
     st.write(f"{datetime.datetime.now()}", context)