Update generate_answer.py

generate_answer.py

@@ -1,8 +1,8 @@
-from transformers import BartTokenizer, BartModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from langchain.prompts import PromptTemplate
 
-tokenizer = BartTokenizer.from_pretrained('facebook/bart-base')
-model = BartModel.from_pretrained('facebook/bart-base')
+tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
+model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
 
 def generate_answer(context):
     prompt_template = PromptTemplate(template="Summarise the following context: {context}",