Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -94,7 +94,7 @@ def generate_text(prompt, system_prompt, temperature, top_p, top_k, repetition_p
|
|
94 |
|
95 |
# Prepare generation kwargs
|
96 |
generation_kwargs = dict(
|
97 |
-
messages,
|
98 |
do_sample=True,
|
99 |
max_new_tokens=max_length,
|
100 |
temperature=temperature,
|
|
|
94 |
|
95 |
# Prepare generation kwargs
|
96 |
generation_kwargs = dict(
|
97 |
+
text_inputs=messages,
|
98 |
do_sample=True,
|
99 |
max_new_tokens=max_length,
|
100 |
temperature=temperature,
|