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
|