File size: 1,093 Bytes
54d1112 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import gradio as gr
from typing import TypedDict
import pandas as pd
class Hit(TypedDict):
cid: str
score: float
text: str
demo: Optional[gr.Interface] = None # Assign your gradio demo to this variable
return_type = List[Hit]
## YOUR_CODE_STARTS_HERE
def retrieve(query: str, topk: int=10) -> return_type:
ranking = bm25_retriever.retrieve(query=query, topk=3)
hits = []
for cid, score in ranking.items():
text = bm25_retriever.index.doc_texts[bm25_retriever.index.cid2docid[cid]]
hits.append({"cid": cid, "score": score, "text": text})
return hits
demo = gr.Interface(
fn=retrieve,
inputs=gr.Textbox(lines=3, placeholder="Enter your query here..."),
outputs="json",
title="CSC BM25 Retriever",
description="Retrieve documents based on the query using CSC BM25 Retriever",
examples=[
["What are the differences between immunodeficiency and autoimmune diseases?"],
["What are the causes of immunodeficiency?"],
["What are the symptoms of immunodeficiency?"],
]
)
## YOUR_CODE_ENDS_HERE
demo.launch() |