joephofhuis commited on
Commit
389ae21
·
verified ·
1 Parent(s): 8291d14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -37,15 +37,15 @@ pipe = pipeline(
37
  tokenizer=tokenizer)
38
 
39
 
40
- def respond(
41
- message,
42
- history: list[tuple[str, str]]
43
- ):
44
  messages = [{"role": "system", "content": SYSTEM_PROMPT}]
 
 
 
 
45
  messages.append({"role": "user", "content": message})
46
 
47
  generation_config = {
48
- 'max_new_tokens':2048,
49
  'temperature':0.7
50
  }
51
  res = pipe(messages,**generation_config)
 
37
  tokenizer=tokenizer)
38
 
39
 
40
+ def respond(message,history):
 
 
 
41
  messages = [{"role": "system", "content": SYSTEM_PROMPT}]
42
+
43
+ for val in history:
44
+ messages.append({"role": val['role'], "content": val['content']})
45
+
46
  messages.append({"role": "user", "content": message})
47
 
48
  generation_config = {
 
49
  'temperature':0.7
50
  }
51
  res = pipe(messages,**generation_config)