Spaces:
Sleeping
Sleeping
add the rank-combined column before using it
Browse files
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 |
-
).
|
| 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()}
|