Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline | |
| model_name = "deepset/roberta-base-squad2" | |
| model = AutoModelForQuestionAnswering.from_pretrained(model_name) | |
| tokenizer = AutoTokenizer.from_pretrained(model_name) | |
| def pred(que,cont): | |
| nlp = pipeline('question-answering', model=model_name, tokenizer=model_name) | |
| QA_input = { | |
| 'question': que, | |
| 'context': cont | |
| } | |
| res = nlp(QA_input) | |
| print(res) | |
| values = list(res.values())[3] | |
| return values | |
| interface = gr.Interface(fn=pred, | |
| inputs=["text", "text"], | |
| outputs="text", | |
| title='Bullet Point') | |
| interface.launch(inline=False) | |