From 5df0bc8005a86851542c25f266d35664edd6b8e5 Mon Sep 17 00:00:00 2001
From: yadonglu <yadonglu@microsoft.com>
Date: Wed, 9 Oct 2024 22:57:41 +0000
Subject: [PATCH] minor:

---
 demo.ipynb     | 4 ++--
 gradio_demo.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/demo.ipynb b/demo.ipynb
index ecdb8cb..f78b52e 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -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"
    ]
diff --git a/gradio_demo.py b/gradio_demo.py
index 55d672d..5522f10 100644
--- a/gradio_demo.py
+++ b/gradio_demo.py
@@ -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 = {