fix
Browse files
app.py
CHANGED
@@ -130,12 +130,6 @@ def chat_with_model(messages):
|
|
130 |
messages[-1]["content"] = output_text
|
131 |
break
|
132 |
|
133 |
-
# Strip prompt from start of generated output
|
134 |
-
if output_text.startswith(prompt_text):
|
135 |
-
output_text = output_text[len(prompt_text):]
|
136 |
-
else:
|
137 |
-
output_text = output_text
|
138 |
-
|
139 |
generated_tokens += 1
|
140 |
if generated_tokens >= max_new_tokens:
|
141 |
break
|
|
|
130 |
messages[-1]["content"] = output_text
|
131 |
break
|
132 |
|
|
|
|
|
|
|
|
|
|
|
|
|
133 |
generated_tokens += 1
|
134 |
if generated_tokens >= max_new_tokens:
|
135 |
break
|