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 |
|