Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def local_generate(
|
|
51 |
)
|
52 |
generated_text = response_output[0]['generated_text']
|
53 |
# Extract the assistant's response
|
54 |
-
result = generated_text[len(prompt):].replace('<think>','Thinking... ')
|
55 |
return result
|
56 |
|
57 |
def respond(
|
|
|
51 |
)
|
52 |
generated_text = response_output[0]['generated_text']
|
53 |
# Extract the assistant's response
|
54 |
+
result = generated_text[len(prompt):].replace('<think>','Thinking... ').replace('</think>','Ended thinking! ')
|
55 |
return result
|
56 |
|
57 |
def respond(
|