Rams901 commited on
Commit
8a76842
·
1 Parent(s): c867eb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def retrieve_thoughts(query, ):
67
  # tier_1 = [doc[0] for doc in docs if ((doc[1] < 1))][:5]
68
  # tier_2 = [doc[0] for doc in docs if ((doc[1] > 0.7)*(doc[1] < 1.5))][10:15]
69
 
70
- return {'tier 1':tier_1_adjusted.loc[:5], 'tier 2': tier_2.loc[:5]}
71
 
72
  def qa_retrieve(query,):
73
 
 
67
  # tier_1 = [doc[0] for doc in docs if ((doc[1] < 1))][:5]
68
  # tier_2 = [doc[0] for doc in docs if ((doc[1] > 0.7)*(doc[1] < 1.5))][10:15]
69
 
70
+ return {'tier 1':tier_1_adjusted.loc[:5], 'tier 2': tier_2.loc[:5]}
71
 
72
  def qa_retrieve(query,):
73