alexpap commited on
Commit
3e73060
1 Parent(s): cbd0f53

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -1
README.md CHANGED
@@ -46,11 +46,20 @@ on relevant data.
46
  ## Use in transformers:
47
 
48
  ```python
49
- from transformers import AutoTokenizer, AutoModelForQuestionAnswering
50
 
51
  tokenizer = AutoTokenizer.from_pretrained("AmazonScience/qanlu", use_auth_token=True)
52
 
53
  model = AutoModelForQuestionAnswering.from_pretrained("AmazonScience/qanlu", use_auth_token=True)
 
 
 
 
 
 
 
 
 
54
  ```
55
 
56
  ## Citation
 
46
  ## Use in transformers:
47
 
48
  ```python
49
+ from transformers import AutoTokenizer, AutoModelForQuestionAnswering, pipeline
50
 
51
  tokenizer = AutoTokenizer.from_pretrained("AmazonScience/qanlu", use_auth_token=True)
52
 
53
  model = AutoModelForQuestionAnswering.from_pretrained("AmazonScience/qanlu", use_auth_token=True)
54
+
55
+ qa_pipeline = pipeline('question-answering', model=model, tokenizer=tokenizer)
56
+
57
+ qa_input = {
58
+ 'context': 'Yes. No. I want a cheap flight to Boston.',
59
+ 'question': 'What is the destination?'
60
+ }
61
+
62
+ answer = qa_pipeline(qa_input)
63
  ```
64
 
65
  ## Citation