MikeMann commited on
Commit
8563685
·
1 Parent(s): 15b06e6

added EvalDataset Generation

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- retrieved_chunks_text = rerankingModel.rerank(query, retrieved_chunks_text, k=15)
 
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)