Rams901 commited on
Commit
4325935
1 Parent(s): 4460317

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def retrieve_thoughts(query, n):
45
  chunks_1 = tier_1.groupby(['_id' ]).apply(lambda x: {f"chunk_{i}": row for i, row in enumerate(x.sort_values('id')[['id', 'score','page_content']].to_dict('records'))}).values
46
  tier_1_adjusted = tier_1.groupby(['_id']).first().reset_index()[['_id', 'title', 'url', 'score']]
47
  tier_1_adjusted['ref'] = range(1, len(tier_1_adjusted) + 1 )
48
- tier_1_adjusted['chunks'] = chunks_1
49
  score = tier_1.groupby(['_id' ]).apply(lambda x: x['score'].mean()).values
50
  tier_1_adjusted['score'] = score
51
  tier_1_adjusted.sort_values("score", inplace = True)
 
45
  chunks_1 = tier_1.groupby(['_id' ]).apply(lambda x: {f"chunk_{i}": row for i, row in enumerate(x.sort_values('id')[['id', 'score','page_content']].to_dict('records'))}).values
46
  tier_1_adjusted = tier_1.groupby(['_id']).first().reset_index()[['_id', 'title', 'url', 'score']]
47
  tier_1_adjusted['ref'] = range(1, len(tier_1_adjusted) + 1 )
48
+ tier_1_adjusted['chunks'] = list(chunks_1)
49
  score = tier_1.groupby(['_id' ]).apply(lambda x: x['score'].mean()).values
50
  tier_1_adjusted['score'] = score
51
  tier_1_adjusted.sort_values("score", inplace = True)