shawon100 commited on
Commit
1416484
·
1 Parent(s): b054d14

Updated app.py with pretrained model

Browse files
Files changed (1) hide show
  1. app.py +15 -8
app.py CHANGED
@@ -1,10 +1,17 @@
1
 
 
2
  import gradio as gr
3
- 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."
4
- question = "Who can replace the equipment?"
5
- gr.Interface.load(
6
- "huggingface/deepset/roberta-base-squad2",
7
- theme="default",
8
- inputs=[gr.inputs.Textbox(lines=7, default=context, label="Context Paragraph"), gr.inputs.Textbox(lines=2, default=question, label="Question")],
9
- outputs=[gr.outputs.Textbox(label="Answer")],
10
- title=None).launch()
 
 
 
 
 
 
 
1
 
2
+ from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
3
  import gradio as gr
4
+
5
+ nlp = pipeline('question-answering', model='deepset/roberta-base-squad2', tokenizer='deepset/roberta-base-squad2')
6
+
7
+ def qnamodel(ques,content):
8
+ question_set = {'question':ques,'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)