| import gradio as gr | |
| import pandas as pd | |
| from query import default_probe, get_retrieval_span_scores | |
| def spaces_probe(question: str): | |
| answers, scores, context = default_probe(question) | |
| answers_text = [answer.text for answer in answers] | |
| d_scores, s_scores = get_retrieval_span_scores(answers) | |
| formatted_result = pd.DataFrame(zip(answers_text, d_scores.tolist(), s_scores.tolist()), columns=[ | |
| "answer", "document score", "span score"]) | |
| formatted_result["position"] = formatted_result.index + 1 | |
| return formatted_result | |
| interface = gr.Interface(spaces_probe, inputs="text", outputs=["dataframe"]) | |
| interface.launch() | |