samarthagarwal23 commited on
Commit
261aa30
·
1 Parent(s): bc84182

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def final_qa_pipeline(file, query):
83
  if len(lvl1) > 0:
84
  fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)
85
  #return (fnl_rank[0]["answer"], str(np.round(100*fnl_rank[0]["score"],2))+"%" , fnl_rank[0]['doc'])
86
- return (fnl_rank[0]["answer"], str(np.round(100*fnl_rank[0]["score"],2))+"%" , print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start_idx'], fnl_rank[0]['end_idx']))
87
  #for fnl_ in fnl_rank:
88
  # print("\n")
89
  # print_colored(fnl_['doc'], fnl_['start'], fnl_['end'])
 
83
  if len(lvl1) > 0:
84
  fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)
85
  #return (fnl_rank[0]["answer"], str(np.round(100*fnl_rank[0]["score"],2))+"%" , fnl_rank[0]['doc'])
86
+ return (fnl_rank[0]["answer"], str(np.round(100*fnl_rank[0]["score"],2))+"%" , print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start'], fnl_rank[0]['end']))
87
  #for fnl_ in fnl_rank:
88
  # print("\n")
89
  # print_colored(fnl_['doc'], fnl_['start'], fnl_['end'])