Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,16 @@ def respond(
|
|
22 |
pipe = pipeline("text-generation", model=model)
|
23 |
|
24 |
yield pipe(
|
25 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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,
|