added EvalDataset Generation
Browse files
app.py
CHANGED
|
@@ -377,7 +377,8 @@ class BSIChatbot:
|
|
| 377 |
#newprint("DBG:retrieved_chunks_text")
|
| 378 |
#newprint(type(retrieved_chunks_text))
|
| 379 |
#new print(retrieved_chunks_text)
|
| 380 |
-
|
|
|
|
| 381 |
#newprint("DBG:retrieved_chunks_text after rerank")
|
| 382 |
#newprint(type(retrieved_chunks_text))
|
| 383 |
#newprint(retrieved_chunks_text)
|
|
|
|
| 377 |
#newprint("DBG:retrieved_chunks_text")
|
| 378 |
#newprint(type(retrieved_chunks_text))
|
| 379 |
#new print(retrieved_chunks_text)
|
| 380 |
+
with torch.no_grad():
|
| 381 |
+
retrieved_chunks_text = rerankingModel.rerank(query, retrieved_chunks_text, k=15)
|
| 382 |
#newprint("DBG:retrieved_chunks_text after rerank")
|
| 383 |
#newprint(type(retrieved_chunks_text))
|
| 384 |
#newprint(retrieved_chunks_text)
|