| 7 | |
| 8 | |
| 9 | def create_retrieve() -> Dict[str, "Component"]: |
| 10 | with gr.Accordion(open=False) as retrieve_tab: |
| 11 | with gr.Row(): |
| 12 | instruction = gr.Textbox(interactive=True) |
| 13 | retrieval_topk = gr.Slider( |
| 14 | minimum=1, maximum=100, value=5, step=1, interactive=True |
| 15 | ) |
| 16 | retrieval_use_fp16 = gr.Checkbox(value=True, interactive=True) |
| 17 | retrieval_pooling_method = gr.Dropdown( |
| 18 | choices=["mean", "pooling", "cls"], interactive=True |
| 19 | ) |
| 20 | with gr.Row(): |
| 21 | query_max_length = gr.Slider( |
| 22 | minimum=1, maximum=2048, value=512, step=1, interactive=True |
| 23 | ) |
| 24 | retrieval_batch_size = gr.Slider( |
| 25 | minimum=1, maximum=1024, value=256, step=1, interactive=True |
| 26 | ) |
| 27 | with gr.Row(): |
| 28 | bm25_backend = gr.Dropdown( |
| 29 | choices=["bm25s", "pyserini"], value="pyserini", interactive=True |
| 30 | ) |
| 31 | use_sentence_transformers = gr.Checkbox(value=False, interactive=True) |
| 32 | with gr.Row(): |
| 33 | save_retrieval_cache = gr.Checkbox(value=False, interactive=True) |
| 34 | use_retrieval_cache = gr.Checkbox(value=False, interactive=True) |
| 35 | retrieval_cache_path = gr.Textbox(interactive=True) |
| 36 | |
| 37 | return dict( |
| 38 | retrieve_tab=retrieve_tab, |
| 39 | instruction=instruction, |
| 40 | retrieval_topk=retrieval_topk, |
| 41 | retrieval_use_fp16=retrieval_use_fp16, |
| 42 | retrieval_pooling_method=retrieval_pooling_method, |
| 43 | query_max_length=query_max_length, |
| 44 | retrieval_batch_size=retrieval_batch_size, |
| 45 | bm25_backend=bm25_backend, |
| 46 | use_sentence_transformers=use_sentence_transformers, |
| 47 | save_retrieval_cache=save_retrieval_cache, |
| 48 | use_retrieval_cache=use_retrieval_cache, |
| 49 | retrieval_cache_path=retrieval_cache_path, |
| 50 | ) |