from transformers import pipeline | |
import streamlit as st | |
question_answerer = pipeline("question-answering", model="distilbert-base-cased-distilled-squad") | |
cont = st.text_area("Give a context:") | |
quest = st.text_input("Your question:") | |
if st.button("OK"): | |
response = question_answerer( | |
question=quest, | |
context=cont | |
) | |
st.write(response) |