Spaces:
Build error
Build error
jadechoghari
commited on
Commit
•
a889d2d
1
Parent(s):
7362041
add readme
Browse files
README.md
CHANGED
@@ -1,3 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
# OmniParser: Screen Parsing tool for Pure Vision Based GUI Agent
|
2 |
|
3 |
<p align="center">
|
|
|
1 |
+
---
|
2 |
+
title: HF's Missing Inference Widget
|
3 |
+
emoji: 💻
|
4 |
+
colorFrom: gray
|
5 |
+
colorTo: gray
|
6 |
+
sdk: gradio
|
7 |
+
pinned: true
|
8 |
+
models:
|
9 |
+
- microsoft/OmniParser
|
10 |
+
|
11 |
+
|
12 |
# OmniParser: Screen Parsing tool for Pure Vision Based GUI Agent
|
13 |
|
14 |
<p align="center">
|
app.py
CHANGED
@@ -14,6 +14,7 @@ from PIL import Image
|
|
14 |
|
15 |
# Model source: https://huggingface.co/microsoft/OmniParser
|
16 |
# gr.load("models/microsoft/OmniParser").launch()
|
|
|
17 |
yolo_model = get_yolo_model(model_path='weights/icon_detect/best.pt')
|
18 |
caption_model_processor = get_caption_model_processor(model_name="florence2", model_name_or_path="weights/icon_caption_florence")
|
19 |
platform = 'pc'
|
|
|
14 |
|
15 |
# Model source: https://huggingface.co/microsoft/OmniParser
|
16 |
# gr.load("models/microsoft/OmniParser").launch()
|
17 |
+
MODEL="microsoft/OmniParser"
|
18 |
yolo_model = get_yolo_model(model_path='weights/icon_detect/best.pt')
|
19 |
caption_model_processor = get_caption_model_processor(model_name="florence2", model_name_or_path="weights/icon_caption_florence")
|
20 |
platform = 'pc'
|