Update app.py

app.py

@@ -74,7 +74,7 @@ def final_qa_pipeline(file, query):
 
     if len(lvl1) > 0:
         fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)
-        return (fnl_rank[0]["answer"], fnl_rank[0]["score"])
+        return (fnl_rank[0]["answer"], np.round(fnl_rank[0]["score"],3), fnl_rank[0]["doc"])
         #for fnl_ in fnl_rank:
         #    print("\n")
         #    print_colored(fnl_['doc'], fnl_['start'], fnl_['end'])
@@ -91,7 +91,7 @@ examples = [
 iface = gr.Interface(
    fn = final_qa_pipeline,
    inputs = [gr.inputs.File(label="input pdf file"), gr.inputs.Textbox(label="Question:")],
-   outputs = [gr.outputs.HTML(label="Answer"), gr.outputs.HTML(label="Score")],
+   outputs = [gr.outputs.HTML(label="Answer"), gr.outputs.HTML(label="Score"), , gr.outputs.HTML(label="doc")],
    examples=examples,
    )
 iface.launch()