Spaces:
Runtime error
Runtime error
changes
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ def generate_text(input_text, history):
|
|
| 29 |
else:
|
| 30 |
input_text_with_history = f"{history[-1][1]}" + "\n"
|
| 31 |
input_text_with_history += f"USER: {input_text}" + "\n" + " ASSISTANT:"
|
| 32 |
-
output = llm(input_text_with_history, max_tokens=
|
| 33 |
"<|prompter|>", "<|endoftext|>", "<|endoftext|> \n",
|
| 34 |
"ASSISTANT:", "USER:", "SYSTEM:"], stream=True
|
| 35 |
)
|
|
@@ -52,7 +52,6 @@ demo = gr.ChatInterface(generate_text,
|
|
| 52 |
"Ты говоришь по русски? Я злой."
|
| 53 |
],
|
| 54 |
cache_examples=True,
|
| 55 |
-
retry_btn="Retry",
|
| 56 |
undo_btn="Undo",
|
| 57 |
clear_btn="Clear")
|
| 58 |
|
|
|
|
| 29 |
else:
|
| 30 |
input_text_with_history = f"{history[-1][1]}" + "\n"
|
| 31 |
input_text_with_history += f"USER: {input_text}" + "\n" + " ASSISTANT:"
|
| 32 |
+
output = llm(input_text_with_history, max_tokens=4096, stop=[
|
| 33 |
"<|prompter|>", "<|endoftext|>", "<|endoftext|> \n",
|
| 34 |
"ASSISTANT:", "USER:", "SYSTEM:"], stream=True
|
| 35 |
)
|
|
|
|
| 52 |
"Ты говоришь по русски? Я злой."
|
| 53 |
],
|
| 54 |
cache_examples=True,
|
|
|
|
| 55 |
undo_btn="Undo",
|
| 56 |
clear_btn="Clear")
|
| 57 |
|