letrunglinh commited on
Commit
649fbe4
1 Parent(s): aaa4456

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ def question_answer(context, question):
6
  model_checkpoint = "letrunglinh/qa_pnc"
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, use_auth_token=AUTH_TOKEN)
9
- model = AutoModelForQuestionAnswering(model_checkpoint)
10
  model = pipeline('question-answering', model=model,
11
  tokenizer=tokenizer, use_auth_token=AUTH_TOKEN)
12
  to_predict = [
 
6
  model_checkpoint = "letrunglinh/qa_pnc"
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, use_auth_token=AUTH_TOKEN)
9
+ model = AutoModelForQuestionAnswering.from_pretrained(model_checkpoint)
10
  model = pipeline('question-answering', model=model,
11
  tokenizer=tokenizer, use_auth_token=AUTH_TOKEN)
12
  to_predict = [