Update app.py
Browse files
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)
|