Rams901 commited on
Commit
fe43fb3
1 Parent(s): 680eb62

Update app.py

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