File size: 1,138 Bytes
3f002dd a57556d 3f002dd 3fd72b4 |
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 36 37 38 |
"""# TASK3: a search-engine demo based on Huggingface space (4 points)
## TASK3.1: create the gradio app (2 point)
Create a gradio app to demo the BM25 search engine index on SciQ. The app should have a single input variable for the query (of type `str`) and a single output variable for the returned ranking (of type `List[Hit]` in the code below).
"""
from typing import TypedDict, Optional, List
import gradio as gr
import copy_of_hw1
from copy_of_hw1 import BM25Retriever
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 hits(query):
Hits = []
bm25_retriever = BM25Retriever(index_dir="output/bm25_index")
retrieved = bm25_retriever.retrieve(query)
for cid in retrieved.keys():
docid = bm25_retriever.index.cid2docid[cid]
doc_text = bm25_retriever.index.doc_texts[docid]
Hits.append(Hit(cid=cid, score=retrieved[cid], text=doc_text))
return Hits
demo = gr.Interface(
fn=hits,
inputs=["text"],
outputs=["text"],
)
## YOUR_CODE_ENDS_HERE
demo.launch() |