Files

 main

/

product_QA.py

Copy path

Blame

Blame

Latest commit

 

History

History

15 lines (10 loc) · 384 Bytes

 main

/

product_QA.py

Top

File metadata and controls

• Code
• Blame

15 lines (10 loc) · 384 Bytes

Raw

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

from transformers import pipeline

model_name = "distilbert-base-cased-distilled-squad"

qa_pipeline = pipeline("question-answering", model=model_name, tokenizer=model_name)

# make a function to get the answer from the question

def get_answer(question, context):

answer = qa_pipeline({

'question': question,

'context': context

})

return answer['answer']