rzanoli commited on
Commit
00cfaee
1 Parent(s): 36e28d7

Minor changes

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -60,6 +60,8 @@ load_vector_store = Chroma(
60
 
61
  retriever = load_vector_store.as_retriever(search_kwargs={"k": 1})
62
 
 
 
63
  #query = "In what context is mentioned Natural Gas Storage Overview?"
64
  #semantic_search = retriever.get_relevant_documents(query)
65
  #print(semantic_search)
@@ -79,6 +81,7 @@ retriever = load_vector_store.as_retriever(search_kwargs={"k": 1})
79
  sample_query = []
80
 
81
  def get_response(input):
 
82
  query = input
83
  chain_type_kwargs = {"prompt": prompt}
84
  qa = RetrievalQA.from_chain_type(
 
60
 
61
  retriever = load_vector_store.as_retriever(search_kwargs={"k": 1})
62
 
63
+ print("retrieval", retriever)
64
+
65
  #query = "In what context is mentioned Natural Gas Storage Overview?"
66
  #semantic_search = retriever.get_relevant_documents(query)
67
  #print(semantic_search)
 
81
  sample_query = []
82
 
83
  def get_response(input):
84
+ print("input", input)
85
  query = input
86
  chain_type_kwargs = {"prompt": prompt}
87
  qa = RetrievalQA.from_chain_type(