-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.py
29 lines (25 loc) · 974 Bytes
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from transformers import pipeline
pipe = pipeline("question-answering", model="deepset/roberta-base-squad2")
def predict(question, context):
result = 'You are welcome'
result = pipe(question=question, context=context)
result = result['answer']
return result
from flask import Flask, request, render_template
app = Flask(__name__)
@app.route('/', methods=['GET', 'POST'])
def index():
if request.method == 'POST':
context = request.form['context']
question = request.form['question']
if context == '':
context = 'None'
if question == '':
question = 'None'
pred = 'None'
pred = predict(question,context)
result = f"result : {pred}"
return render_template('index.html', result=result,context=context, question=question)
return render_template('index.html',context='', question='')
if __name__ == '__main__':
app.run(debug=True)