ppsingh commited on
Commit
e878a02
1 Parent(s): 1593f40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -49,10 +49,10 @@ async def chat(query,history,sources,reports):
49
  if len(reports) == 0:
50
  reports = []
51
 
52
- if sources == ["ABC"]:
53
- vectorstore = vectorstores["ABC"]
54
  else:
55
- vectorstore = vectorstores["XYZ"]
56
 
57
  # get context
58
  context_retrieved_lst = []
 
49
  if len(reports) == 0:
50
  reports = []
51
 
52
+ if sources == ["Ministry"]:
53
+ vectorstore = vectorstores["MWTS"]
54
  else:
55
+ vectorstore = vectorstores["Consolidated"]
56
 
57
  # get context
58
  context_retrieved_lst = []