Saurabhz commited on
Commit
e36d3cb
·
1 Parent(s): f78da49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -5,15 +5,15 @@ from transformers import pipeline
5
  def question_answer(context, question):
6
  model_checkpoint = "huggingface-course/bert-finetuned-squad"
7
  model = pipeline("question-answering", model=model_checkpoint)
8
- to_predict = [
9
  {
10
- "context": context,
11
- "qas": [
12
- {
13
- "question": question,
14
- "id": "0",
15
- }
16
- ],
17
  }
18
  ]
19
  answers, probabilities = model.predict(to_predict)
 
5
  def question_answer(context, question):
6
  model_checkpoint = "huggingface-course/bert-finetuned-squad"
7
  model = pipeline("question-answering", model=model_checkpoint)
8
+ to_predict = [
9
  {
10
+ "context": context,
11
+ "qas": [
12
+ {
13
+ "question": question,
14
+ "id": "0",
15
+ }
16
+ ],
17
  }
18
  ]
19
  answers, probabilities = model.predict(to_predict)