Update app.py
Browse files
app.py
CHANGED
@@ -86,7 +86,7 @@ def detect_combined(image):
|
|
86 |
url = "https://api.landing.ai/v1/tools/text-to-object-detection"
|
87 |
headers = {"Authorization": f"Basic {COUNTGD_API_KEY}"}
|
88 |
competitor_boxes = []
|
89 |
-
COUNTGD_PROMPTS = ["cans", "bottle", "
|
90 |
|
91 |
for prompt in COUNTGD_PROMPTS:
|
92 |
with open(temp_path, "rb") as f:
|
@@ -134,7 +134,7 @@ def detect_combined(image):
|
|
134 |
cv2.imwrite(output_path, img)
|
135 |
|
136 |
# Buat result text untuk count produk Nestlé per class dan total keseluruhan
|
137 |
-
result_text = "Produk
|
138 |
for class_name, count in nestle_class_count.items():
|
139 |
result_text += f"{class_name}: {count}\n"
|
140 |
total_nestle = sum(nestle_class_count.values())
|
|
|
86 |
url = "https://api.landing.ai/v1/tools/text-to-object-detection"
|
87 |
headers = {"Authorization": f"Basic {COUNTGD_API_KEY}"}
|
88 |
competitor_boxes = []
|
89 |
+
COUNTGD_PROMPTS = ["cans", "bottle", "beverage"]
|
90 |
|
91 |
for prompt in COUNTGD_PROMPTS:
|
92 |
with open(temp_path, "rb") as f:
|
|
|
134 |
cv2.imwrite(output_path, img)
|
135 |
|
136 |
# Buat result text untuk count produk Nestlé per class dan total keseluruhan
|
137 |
+
result_text = "Produk Nestlé:\n"
|
138 |
for class_name, count in nestle_class_count.items():
|
139 |
result_text += f"{class_name}: {count}\n"
|
140 |
total_nestle = sum(nestle_class_count.values())
|