Update generate_answers.py

generate_answers.py

@@ -1,9 +1,7 @@
-from transformers import BertForQuestionAnswering, BertTokenizer
-import torch
+from transformers import AutoTokenizer, AutoModelForQuestionAnswering
 
-model_name = "bert-large-uncased-whole-word-masking-finetuned-squad"
-tokenizer = BertTokenizer.from_pretrained(model_name)
-model = BertForQuestionAnswering.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained("Vardan-verma/Question_Answering_model_finetuned_on_bert")
+model = AutoModelForQuestionAnswering.from_pretrained("Vardan-verma/Question_Answering_model_finetuned_on_bert")
 
 def get_answer(question, context):
     inputs = tokenizer(question, context, return_tensors="pt", truncation=True, max_length=512)