Update generate_answer.py

generate_answer.py

@@ -8,7 +8,7 @@ def generate_answer(context):
     prompt_template = PromptTemplate(template="Summarise the following context: {context}",
                                      input_variables=["context"], output_variables=["answer"])
     # Model loading
-    format_prompt = prompt_template.format(question=question)
+    format_prompt = prompt_template.format(context=context)
     encoded_input = tokenizer(format_prompt, return_tensors='pt')
     # Run the model
     output = model.generate(**encoded_input)  # Use generate method for text generation