Spaces:
Running
Running
being more selective with the columns that print in the output
Browse files
app.py
CHANGED
|
@@ -44,8 +44,9 @@ 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 |
-
|
|
|
|
| 49 |
|
| 50 |
#return {"request": request, "results": str(res_combined)}
|
| 51 |
#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 |
+
(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()}
|