Where I can get 'weights/omniparser/icon_caption_blip2/best.pt'

[binchen1998]

Hi,

When I ran the example, it said:

FileNotFoundError: [Errno 2] No such file or directory: 'weights/omniparser/icon_caption_blip2/best.pt'
How to get these files?

Thanks a lot.

[runshengdu]

You need to create the file path manually, and go to huggingface to download the best.pt

[rwtnb]

Clone the HF repo and change the yolo model path in gradio_demo.py

yolo_model = get_yolo_model(model_path='<path>/best.pt')

[binchen1998]

Clone the HF repo and change the yolo model path in gradio_demo.py

yolo_model = get_yolo_model(model_path='<path>/best.pt')

Thanks for your reply!

But I noticed that the best.pt in HF repo is under icon_detect/ directory, but the code refers to icon_caption_blip2/ directory, is the best.pt in icon_detect/ usable?

[yadong-lu]

Hi @binchen1998 , we removed the .pt file with model safetensor file in the HF repo. Please try the updated readme, and see if you have any issue.

[abrichr]

See #52 for a download.py script that automates downloading the weights (used in this Dockerfile in the same PR).