| import json |
| import streamlit as st |
| import requests as req |
|
|
| |
| |
|
|
|
|
| st.title('Question answering help desk application') |
|
|
|
|
| |
| st.subheader('1. A simple question') |
| |
|
|
|
|
| WIKI_URL = 'https://en.wikipedia.org/w/api.php' |
| WIKI_QUERY = "?format=json&action=query&prop=extracts&explaintext=1" |
| WIKI_BERT = "&titles=BERT_(language_model)" |
| WIKI_METHOD = 'GET' |
|
|
| response = req.request(WIKI_METHOD, f'{WIKI_URL}{WIKI_QUERY}{WIKI_BERT}') |
| resp_json = json.loads(response.content.decode("utf-8")) |
| wiki_bert = resp_json['query']['pages']['62026514']['extract'] |
| paragraph = wiki_bert |
|
|
| written_passage = st.text_area( |
| 'Paragraph used for QA (you can also edit, or copy/paste new content)', |
| paragraph, |
| height=250 |
| ) |
| if written_passage: |
| paragraph = written_passage |
|
|
| question = 'How many languages does bert understand?' |
| written_question = st.text_input( |
| 'Question used for QA (you can also edit, and experiment with the answers)', |
| question |
| ) |
| if written_question: |
| question = written_question |
|
|
| QA_URL = "https://api-inference.huggingface.co/models/deepset/roberta-base-squad2" |
| QA_METHOD = 'POST' |
|
|
|
|
| if st.button('Run QA inference (get answer prediction)'): |
| if paragraph and question: |
| inputs = {'question': question, 'context': paragraph} |
| payload = json.dumps(inputs) |
| prediction = req.request(QA_METHOD, QA_URL, data=payload) |
| answer = json.loads(prediction.content.decode("utf-8")) |
| answer_span = answer["answer"] |
| answer_score = answer["score"] |
| st.write(f'Answer: **{answer_span}**') |
| start_par = max(0, answer["start"]-86) |
| stop_para = min(answer["end"]+90, len(paragraph)) |
| answer_context = paragraph[start_par:stop_para].replace(answer_span, f'**{answer_span}**') |
| st.write(f'Answer context (and score): ... _{answer_context}_ ... (score: {format(answer_score, ".3f")})') |
| st.write(f'Answer JSON: ') |
| st.write(answer) |
| else: |
| st.write('Write some passage of text and a question') |
| st.stop() |
