Update generator.py

generator.py

@@ -2,9 +2,9 @@ from transformers import pipeline
 
 
 
-def format(self, context) -> str:
+def format(self, text) -> str:
     system_message = f"You are an expert in writing novels. You have won many prices."
-    user_message = f"Please assist based on the following context: {context}"
+    user_message = f"Please assist based on the following context: {text}"
     prompt = f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\n{system_message}<|eot_id|>" 
     prompt += f"<|start_header_id|>user<|end_header_id|>\n\n{user_input}<|eot_id|>"
     prompt += "<|start_header_id|>assistant<|end_header_id|>\n\n"