File size: 397 Bytes
c70f210
 
 
5b7bd18
 
c70f210
 
 
ac93bcc
c70f210
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import streamlit as st
from transformers import pipeline

model_name = 'fungi/deberta-v3-xsmall-squad2'
qa_pipe = pipeline('question-answering', model=model_name, tokenizer=model_name)
context = st.text_area('provide context')
question = st.text_area('ask your question')

if question and context:
    out = qa_pipe(dict(
        question=question,
        context=context
    ))
    st.json(out)