.split("### Email:")

llama.py

@@ -44,6 +44,8 @@ def generate_llama_summary(email, model, tokenizer, prompt_template):
         summary = summary.split("### Summary:")[-1].strip()
     else:
         summary = "Error: Could not extract summary"
+    if "### Email:" in summary:
+        summary = summary.split("### Email:")[0].strip()
     return summary