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
demo.ipynb
View File

@@ -385,7 +385,7 @@
"from PIL import Image\n",
"device = 'cuda'\n",
"\n",
"som_model = get_yolo_model(model_path='weights/omniparser/weights/best.pt')\n",
"som_model = get_yolo_model(model_path='weights/omniparser/icon_detect/best.pt')\n",
"som_model.to(device)\n"
]
},
@@ -404,7 +404,7 @@
],
"source": [
"\n",
"caption_model_processor = get_caption_model_processor(model_name_or_path=\"weights/omniparser/blipv2_ui_merge\", device=device)\n",
"caption_model_processor = get_caption_model_processor(model_name_or_path=\"weights/omniparser/icon_caption_blip2\", device=device)\n",
"\n",
"\n"
]

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 = {