Spaces:

samarthagarwal23
/

QuestionAnswering_on_annual_reports

Runtime error

samarthagarwal23 commited on Jan 30, 2022

Commit

ad9a1e9

•

1 Parent(s): 5da1616

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,7 +93,7 @@ def final_qa_pipeline(file, query):
         fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)
         #return (fnl_rank[0]["answer"], str(np.round(100*fnl_rank[0]["score"],2))+"%" , fnl_rank[0]['doc'])
         #return (print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start'], fnl_rank[0]['end']), str(np.round(100*fnl_rank[0]["score"],2))+"%"
-        top1 = print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start'], fnl_rank[0]['end'], str(np.round(100*fnl_rank[0]["score"],2))+"%")
         if len(lvl1)>1:
             top2 = print_colored(fnl_rank[1]['doc'], fnl_rank[1]['start'], fnl_rank[1]['end'], str(np.round(100*fnl_rank[1]["score"],2))+"%")
         else:

         fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)
         #return (fnl_rank[0]["answer"], str(np.round(100*fnl_rank[0]["score"],2))+"%" , fnl_rank[0]['doc'])
         #return (print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start'], fnl_rank[0]['end']), str(np.round(100*fnl_rank[0]["score"],2))+"%"
+        top1 = print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start'], fnl_rank[0]['end'], str(np.round(100*fnl_rank[0]["score"],1))+"%")
         if len(lvl1)>1:
             top2 = print_colored(fnl_rank[1]['doc'], fnl_rank[1]['start'], fnl_rank[1]['end'], str(np.round(100*fnl_rank[1]["score"],2))+"%")
         else: