Hayk Arutyunyan commited on
Commit ·
43099ab
1
Parent(s): d456ca5
Fix app.py
Browse files
app.py
CHANGED
|
@@ -25,18 +25,18 @@ def draw_boxes(img_rgb, sensors):
|
|
| 25 |
# ================================
|
| 26 |
# Основная функция демо
|
| 27 |
# ================================
|
| 28 |
-
def hf_process(
|
| 29 |
|
| 30 |
cfg = load_config("configs/config.yaml")
|
| 31 |
|
| 32 |
# Запуск пайплайна
|
| 33 |
-
result = process_single_image(
|
| 34 |
|
| 35 |
title = result["title"]
|
| 36 |
sensors = result["sensors"]
|
| 37 |
|
| 38 |
# --- визуализация сенсоров ---
|
| 39 |
-
boxed = draw_boxes(
|
| 40 |
|
| 41 |
# --- DataFrame сенсоров ---
|
| 42 |
df = pd.DataFrame(sensors)[["text", "score", "x", "y", "w", "h"]]
|
|
|
|
| 25 |
# ================================
|
| 26 |
# Основная функция демо
|
| 27 |
# ================================
|
| 28 |
+
def hf_process(img):
|
| 29 |
|
| 30 |
cfg = load_config("configs/config.yaml")
|
| 31 |
|
| 32 |
# Запуск пайплайна
|
| 33 |
+
result = process_single_image(img)
|
| 34 |
|
| 35 |
title = result["title"]
|
| 36 |
sensors = result["sensors"]
|
| 37 |
|
| 38 |
# --- визуализация сенсоров ---
|
| 39 |
+
boxed = draw_boxes(img, sensors)
|
| 40 |
|
| 41 |
# --- DataFrame сенсоров ---
|
| 42 |
df = pd.DataFrame(sensors)[["text", "score", "x", "y", "w", "h"]]
|