sjrhuschlee
/

flan-t5-base-squad2

Question Answering

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

sjrhuschlee commited on Jun 17, 2023

Commit

606792c

·

1 Parent(s): 914343a

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -55,7 +55,9 @@ res = nlp(qa_input)
 # {'score': 0.980, 'start': 30, 'end': 37, 'answer': ' London'}
 # b) Load model & tokenizer
-model = AutoModelForQuestionAnswering.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 question = f'{tokenizer.cls_token}Where do I live?'  # '<cls>Where do I live?'

 # {'score': 0.980, 'start': 30, 'end': 37, 'answer': ' London'}
 # b) Load model & tokenizer
+model = AutoModelForQuestionAnswering.from_pretrained(
+  model_name, trust_remote_code=True
+)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 question = f'{tokenizer.cls_token}Where do I live?'  # '<cls>Where do I live?'