Spaces:
Runtime error
Runtime error
Updated app.py with pretrained model
Browse files
app.py
CHANGED
@@ -4,13 +4,15 @@ import gradio as gr
|
|
4 |
|
5 |
nlp = pipeline('question-answering', model='deepset/roberta-base-squad2', tokenizer='deepset/roberta-base-squad2')
|
6 |
|
7 |
-
def qnamodel(question,
|
8 |
-
|
|
|
|
|
9 |
results = nlp(question_set)
|
10 |
return results['answer']
|
11 |
|
12 |
interface = gr.Interface(fn=qnamodel,
|
13 |
-
inputs=["
|
14 |
outputs="text",
|
15 |
title='Context Question Answering')
|
16 |
|
|
|
4 |
|
5 |
nlp = pipeline('question-answering', model='deepset/roberta-base-squad2', tokenizer='deepset/roberta-base-squad2')
|
6 |
|
7 |
+
def qnamodel(question,context):
|
8 |
+
context = "This procedure is used to guide technician to replace equipment EQ1. The procedure is valid in normal operation. There is different procedure for emergency situations. The replacement of EQ1 may be done by SE, MT, or MS."
|
9 |
+
question = "Who can replace the equipment?"
|
10 |
+
question_set = {'context':context,'question':question}
|
11 |
results = nlp(question_set)
|
12 |
return results['answer']
|
13 |
|
14 |
interface = gr.Interface(fn=qnamodel,
|
15 |
+
inputs=[gr.inputs.Textbox(lines=7, default=context, label="Context Paragraph"), gr.inputs.Textbox(lines=2, default=question, label="Question")],
|
16 |
outputs="text",
|
17 |
title='Context Question Answering')
|
18 |
|