test / app.py
123Chr1s's picture
Update app.py
f54a6f9 verified
raw
history blame contribute delete
682 Bytes
from transformers import pipeline
import streamlit as st
st.title("Question Answering with XLM-RoBERTa")
# Load the question-answering pipeline with the new model
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.")