Update app.py

app.py

@@ -67,11 +67,12 @@ def main():
             #PATH = 'model/'
             #llm = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-medium")
             llm = huggingface_hub.HuggingFaceHub(repo_id="openai-community/gpt2-medium",model_kwargs={"temperature":1.0, "max_length":500})
-            docs = vector_store.similarity_search(query=query, k=1)
-            st.write(docs)
+            #docs = vector_store.similarity_search(query=query, k=1)
+            #st.write(docs)
             #chain = load_qa_chain(llm=llm, chain_type="stuff")
             #response = chain.run(input_documents=docs, question=query)
             retriever=vector_store.as_retriever(search_kargs={"k":1})
+            st.write(retriever)
             chain = RetrievalQA.from_chain_type(llm=llm,chain_type="stuff",retriever=retriever)
             response = chain.run(query)
             st.write(response)