Update generator.py

generator.py

@@ -6,7 +6,7 @@ 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|>" 
-    prompt += f"<|start_header_id|>user<|end_header_id|>\n\n{user_input}<|eot_id|>"
+    prompt += f"<|start_header_id|>user<|end_header_id|>\n\n{user_message}<|eot_id|>"
     prompt += "<|start_header_id|>assistant<|end_header_id|>\n\n"
     return prompt