Rams901 commited on
Commit
6da8cb0
1 Parent(s): 49a6ecc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,7 +74,7 @@ def retrieve_thoughts(query, n):
74
  # tier_1 = [doc[0] for doc in docs if ((doc[1] < 1))][:5]
75
  # tier_2 = [doc[0] for doc in docs if ((doc[1] > 0.7)*(doc[1] < 1.5))][10:15]
76
 
77
- return {'tier 1':tier_1_adjusted, 'tier 2': tier_2.loc[:5]}
78
 
79
  def qa_retrieve(query, llm):
80
 
@@ -95,7 +95,7 @@ def qa_retrieve(query, llm):
95
  mp_docs = thoughts
96
 
97
  tier_1 = thoughts['tier 1']
98
- tier_2 = thoughts['tier 2']
99
 
100
  reference = tier_1[['ref', 'url', 'title', 'content','score']].to_dict('records')
101
 
 
74
  # tier_1 = [doc[0] for doc in docs if ((doc[1] < 1))][:5]
75
  # tier_2 = [doc[0] for doc in docs if ((doc[1] > 0.7)*(doc[1] < 1.5))][10:15]
76
 
77
+ return {'tier 1':tier_1_adjusted, }
78
 
79
  def qa_retrieve(query, llm):
80
 
 
95
  mp_docs = thoughts
96
 
97
  tier_1 = thoughts['tier 1']
98
+ # tier_2 = thoughts['tier 2']
99
 
100
  reference = tier_1[['ref', 'url', 'title', 'content','score']].to_dict('records')
101