Skip to content

Commit

Permalink
update prompt
Browse files Browse the repository at this point in the history
  • Loading branch information
juancwu committed Nov 24, 2024
1 parent 5919b1c commit cd1c039
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion backend/generate_answer.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ def get_device():

def generate_initial_note(page_content, model, tokenizer):
device = "cuda" if torch.cuda.is_available() else "cpu"
inputs = tokenizer(f"Generate notes for the given content: {page_content}", return_tensors="pt").to(device)
inputs = tokenizer(f"Generate notes for the given content: {page_content}\n\nDo not include the original content in your answer.", return_tensors="pt").to(device)
outputs = model.generate(
**inputs,
max_length=2048,
Expand Down

0 comments on commit cd1c039

Please sign in to comment.