Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -46,8 +46,8 @@ def generate_response(prompt, max_length=124):
|
|
| 46 |
print("Raw output:", outputs)
|
| 47 |
# Extract just the assistant's response
|
| 48 |
try:
|
| 49 |
-
#
|
| 50 |
-
message_list =
|
| 51 |
# Get the last message (assistant's response)
|
| 52 |
assistant_message = message_list[-1]
|
| 53 |
if assistant_message["role"] == "assistant":
|
|
|
|
| 46 |
print("Raw output:", outputs)
|
| 47 |
# Extract just the assistant's response
|
| 48 |
try:
|
| 49 |
+
# outputs[0]["generated_text"] is already a list of messages
|
| 50 |
+
message_list = outputs[0]["generated_text"]
|
| 51 |
# Get the last message (assistant's response)
|
| 52 |
assistant_message = message_list[-1]
|
| 53 |
if assistant_message["role"] == "assistant":
|