wbrooks commited on
Commit
861d14f
·
1 Parent(s): 9fbd1cf

add the rank-combined column before using it

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -44,9 +44,10 @@ def search(q: str = Query(..., description="Search query")):
44
 
45
  res_combined = joined.with_columns(
46
  (0.7 * pl.col("rank-sbert") + 0.3 * pl.col("rank-tfidf")).alias("rank-combined"),
47
- pl.col("file").str.strip_prefix(path_prefix).alias("file"),
 
48
  (20.0 / pl.col('rank-combined')).round(2).alias('confidence')
49
- ).sort("rank-combined").select(['file', 'confidence'])
50
 
51
  #return {"request": request, "results": str(res_combined)}
52
  #return {"request": request, "results": res_combined.to_dicts()}
 
44
 
45
  res_combined = joined.with_columns(
46
  (0.7 * pl.col("rank-sbert") + 0.3 * pl.col("rank-tfidf")).alias("rank-combined"),
47
+ pl.col("file").str.strip_prefix(path_prefix).alias("file")
48
+ ).sort("rank-combined").with_columns(
49
  (20.0 / pl.col('rank-combined')).round(2).alias('confidence')
50
+ ).select(['file', 'confidence'])
51
 
52
  #return {"request": request, "results": str(res_combined)}
53
  #return {"request": request, "results": res_combined.to_dicts()}