Nathan Slaughter commited on
Commit
c87e011
·
1 Parent(s): 135612a

Update prompt

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,10 +48,10 @@ def extract_action_items(transcript: str) -> str:
48
  )
49
  tokenizer = AutoTokenizer.from_pretrained(model_name)
50
 
51
- prompt = "Give me a short introduction to large language model."
52
  messages = [
53
  {"role": "system", "content": "You are the perfect action item extraction system."},
54
- {"role": "user", "content": prompt}
 
55
  ]
56
  text = tokenizer.apply_chat_template(
57
  messages,
 
48
  )
49
  tokenizer = AutoTokenizer.from_pretrained(model_name)
50
 
 
51
  messages = [
52
  {"role": "system", "content": "You are the perfect action item extraction system."},
53
+ {"role": "user", "content": "Extract the action items from the voice note transcript using the following format:\n\n- [title 1]: [description 1]\n- [title 2]: [description 2]\n..."},
54
+ {"role": "user", "content": f"transcript: {transcript}"}
55
  ]
56
  text = tokenizer.apply_chat_template(
57
  messages,