Update generator.py

generator.py

@@ -2,7 +2,7 @@ from transformers import pipeline
 
 
 
-def format(self, text) -> str:
+def format( 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: {text}"
     prompt = f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\n{system_message}<|eot_id|>"