Spaces:
Runtime error
Runtime error
Updated app.py with pretrained model
Browse files
app.py
CHANGED
@@ -4,14 +4,14 @@ import gradio as gr
|
|
4 |
|
5 |
nlp = pipeline('question-answering', model='deepset/roberta-base-squad2', tokenizer='deepset/roberta-base-squad2')
|
6 |
|
7 |
-
def qnamodel(
|
8 |
-
question_set = {'question':
|
9 |
results = nlp(question_set)
|
10 |
return results['answer']
|
11 |
|
12 |
interface = gr.Interface(fn=qnamodel,
|
13 |
inputs=["text","text"],
|
14 |
-
|
15 |
title='Context Question Answering')
|
16 |
|
17 |
interface.launch(inline=False)
|
|
|
4 |
|
5 |
nlp = pipeline('question-answering', model='deepset/roberta-base-squad2', tokenizer='deepset/roberta-base-squad2')
|
6 |
|
7 |
+
def qnamodel(question,content):
|
8 |
+
question_set = {'question':question,'context':content}
|
9 |
results = nlp(question_set)
|
10 |
return results['answer']
|
11 |
|
12 |
interface = gr.Interface(fn=qnamodel,
|
13 |
inputs=["text","text"],
|
14 |
+
outputs="text",
|
15 |
title='Context Question Answering')
|
16 |
|
17 |
interface.launch(inline=False)
|