Update app.py
Browse files
app.py
CHANGED
|
@@ -121,7 +121,7 @@ def respond(
|
|
| 121 |
|
| 122 |
output = pipe(text=messages, max_new_tokens=max_new_tokens)
|
| 123 |
|
| 124 |
-
|
| 125 |
|
| 126 |
|
| 127 |
"""
|
|
|
|
| 121 |
|
| 122 |
output = pipe(text=messages, max_new_tokens=max_new_tokens)
|
| 123 |
|
| 124 |
+
return output[0]["generated_text"][-1]["content"]
|
| 125 |
|
| 126 |
|
| 127 |
"""
|