Spaces:
Running
on
Zero
Running
on
Zero
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":
|