Rams901 commited on
Commit
8474d86
1 Parent(s): e7f116b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -59,6 +59,8 @@ def retrieve_thoughts(query, ):
59
  tier_1_adjusted = tier_1.groupby(['title', 'url', ]).first().reset_index()[[ 'title', 'url']]
60
  tier_1_adjusted['ref'] = range(1, len(tier_1_adjusted) + 1 )
61
  tier_1_adjusted['content'] = chunks_1
 
 
62
 
63
  chunks_2 = tier_2.groupby(['title', 'url', ]).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values)).values
64
  tier_2_adjusted = tier_2.groupby(['title', 'url', ]).first().reset_index()[[ 'title', 'url']]
 
59
  tier_1_adjusted = tier_1.groupby(['title', 'url', ]).first().reset_index()[[ 'title', 'url']]
60
  tier_1_adjusted['ref'] = range(1, len(tier_1_adjusted) + 1 )
61
  tier_1_adjusted['content'] = chunks_1
62
+ tier_1_adjusted['score'] = score
63
+
64
 
65
  chunks_2 = tier_2.groupby(['title', 'url', ]).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values)).values
66
  tier_2_adjusted = tier_2.groupby(['title', 'url', ]).first().reset_index()[[ 'title', 'url']]