ayan4m1 commited on
Commit
70872ee
·
verified ·
1 Parent(s): 6a4cab0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -22,7 +22,16 @@ def respond(
22
  pipe = pipeline("text-generation", model=model)
23
 
24
  yield pipe(
25
- message,
 
 
 
 
 
 
 
 
 
26
  max_new_tokens=max_new_tokens,
27
  do_sample=True,
28
  temperature=temperature,
 
22
  pipe = pipeline("text-generation", model=model)
23
 
24
  yield pipe(
25
+ [
26
+ {
27
+ "role": "system",
28
+ "content": system_prompt
29
+ },
30
+ {
31
+ "role": "user",
32
+ "content": message
33
+ }
34
+ ],
35
  max_new_tokens=max_new_tokens,
36
  do_sample=True,
37
  temperature=temperature,