Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,9 @@ def detect_fruit(image):
|
|
| 32 |
|
| 33 |
detected_fruit = None
|
| 34 |
|
|
|
|
| 35 |
for det in detections:
|
|
|
|
| 36 |
label = model.names[int(det[5])] # название фрукта
|
| 37 |
if label in fruits_data:
|
| 38 |
detected_fruit = label
|
|
@@ -86,7 +88,7 @@ image_input = gr.Image(
|
|
| 86 |
)
|
| 87 |
weight_input = gr.Number(label="Вес (кг)")
|
| 88 |
image_output = gr.Image(
|
| 89 |
-
label="Распознанный
|
| 90 |
type="numpy",
|
| 91 |
width=640,
|
| 92 |
height=480
|
|
@@ -102,6 +104,6 @@ gr.Interface(
|
|
| 102 |
fn=gradio_interface,
|
| 103 |
inputs=[image_input, weight_input],
|
| 104 |
outputs=[image_output, receipt_output],
|
| 105 |
-
title="Определение
|
| 106 |
description="Загрузите изображение, введите вес и получите чек"
|
| 107 |
).launch()
|
|
|
|
| 32 |
|
| 33 |
detected_fruit = None
|
| 34 |
|
| 35 |
+
print(model.names)
|
| 36 |
for det in detections:
|
| 37 |
+
print(det)
|
| 38 |
label = model.names[int(det[5])] # название фрукта
|
| 39 |
if label in fruits_data:
|
| 40 |
detected_fruit = label
|
|
|
|
| 88 |
)
|
| 89 |
weight_input = gr.Number(label="Вес (кг)")
|
| 90 |
image_output = gr.Image(
|
| 91 |
+
label="Распознанный товар",
|
| 92 |
type="numpy",
|
| 93 |
width=640,
|
| 94 |
height=480
|
|
|
|
| 104 |
fn=gradio_interface,
|
| 105 |
inputs=[image_input, weight_input],
|
| 106 |
outputs=[image_output, receipt_output],
|
| 107 |
+
title="Определение товара и создание чека",
|
| 108 |
description="Загрузите изображение, введите вес и получите чек"
|
| 109 |
).launch()
|