Spaces:
Sleeping
Sleeping
from transformers import AutoModelForQuestionAnswering,AutoTokenizer,pipeline | |
import streamlit as st | |
import ast | |
model_name="deepset/roberta-base-squad2" | |
pipe=pipeline('question-answering',model=model_name,tokenizer=model_name) | |
def answer_question(question,context): | |
text="{"+"'question': '"+question + "','context':'"+context+"'}" | |
di= ast.literal_eval(text) | |
response= pipe(di) | |
return response | |
st.title("Q&A App") | |
context = st.text_area("Enter the context:") | |
question = st.text_input("Enter your question:") | |
if st.button("Get Answer"): | |
answer = answer_question(context, question) | |
st.write(f"Answer: {answer}") | |