Dom/OmniParser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor:

Browse Source
This commit is contained in:
yadonglu
2024-10-09 22:57:41 +00:00
parent 741af789a8
commit 5df0bc8005
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gradio_demo.py
View File

@@ -12,8 +12,8 @@ from utils import check_ocr_box, get_yolo_model, get_caption_model_processor, ge
import torch
from PIL import Image
yolo_model = get_yolo_model(model_path='weights/omniparser/weights/best.pt')
caption_model_processor = get_caption_model_processor(model_name_or_path="weights/omniparser/blipv2_ui_merge", device='cuda')
yolo_model = get_yolo_model(model_path='weights/omniparser/icon_caption_blip2/best.pt')
caption_model_processor = get_caption_model_processor(model_name_or_path="weights/omniparser/icon_caption_blip2", device='cuda')
platform = 'pc'
if platform == 'pc':
draw_bbox_config = {