github/autoMate
1
0
Fork 0
mirror of https://github.com/yuruotong1/autoMate.git synced 2026-03-22 13:07:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

更新message传递

Browse Source
This commit is contained in:
yuruo
2025-03-04 14:32:24 +08:00
parent 9338b8411d
commit 9470235357
7 changed files with 9 additions and 539 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
util/omniparser.py
View File

@@ -11,7 +11,7 @@ class Omniparser(object):
self.som_model = get_yolo_model(model_path=config['som_model_path'])
self.caption_model_processor = get_caption_model_processor(model_name=config['caption_model_name'], model_name_or_path=config['caption_model_path'], device=device)
print('Omniparser initialized!!!')
print('Omniparser initialized!')
def parse(self, image_base64: str):
image_bytes = base64.b64decode(image_base64)