Spaces:
Runtime error
Runtime error
update cap
Browse files
multimodal/open_flamingo/chat/conversation.py
CHANGED
|
@@ -458,7 +458,7 @@ class Chat:
|
|
| 458 |
# f"### Assistant: {tokenizer.decode(outputs[0, input_ids.shape[1]:], skip_special_tokens=True).strip()}")
|
| 459 |
output_text = self.tokenizer.decode(text_outputs[0])
|
| 460 |
print(output_text)
|
| 461 |
-
output_text = re.findall(r'Assistant:(.+)', output_text)
|
| 462 |
print(output_text)
|
| 463 |
|
| 464 |
return output_text, out_image
|
|
|
|
| 458 |
# f"### Assistant: {tokenizer.decode(outputs[0, input_ids.shape[1]:], skip_special_tokens=True).strip()}")
|
| 459 |
output_text = self.tokenizer.decode(text_outputs[0])
|
| 460 |
print(output_text)
|
| 461 |
+
output_text = re.findall(r'Assistant:(.+)', output_text)[-1]
|
| 462 |
print(output_text)
|
| 463 |
|
| 464 |
return output_text, out_image
|