|
from transformers import pipeline |
|
import streamlit as st |
|
|
|
st.title("Question Answering with XLM-RoBERTa") |
|
|
|
|
|
with st.spinner('Loading model...'): |
|
qa_pipeline = pipeline("question-answering", model="IProject-10/xlm-roberta-base-finetuned-squad2") |
|
|
|
context = st.text_area("Context", "Provide the context here...") |
|
question = st.text_input("Question", "Ask your question here...") |
|
|
|
if st.button("Get Answer"): |
|
if context and question: |
|
result = qa_pipeline(question=question, context=context) |
|
st.write(f"Answer: {result['answer']}") |
|
else: |
|
st.write("Please provide both context and a question.") |