Spaces:
Runtime error
Runtime error
updated roboflow model to improve accuracy
Browse files
app.py
CHANGED
@@ -270,14 +270,15 @@ def fetchImage():
|
|
270 |
file_name, model_id="car-damage-detection-krsix/1")
|
271 |
print(result)
|
272 |
threshold = 0.3 # Adjust this as needed
|
273 |
-
filtered_labels = [item
|
274 |
for item in result["predictions"] if item["confidence"] >= threshold]
|
275 |
|
276 |
print(filtered_labels)
|
277 |
labels = filtered_labels
|
278 |
print(labels)
|
279 |
for class_ in labels:
|
280 |
-
total_cost += repair_costs.get(class_,
|
|
|
281 |
result = {"total_cost": total_cost}
|
282 |
print(result)
|
283 |
|
|
|
270 |
file_name, model_id="car-damage-detection-krsix/1")
|
271 |
print(result)
|
272 |
threshold = 0.3 # Adjust this as needed
|
273 |
+
filtered_labels = [item
|
274 |
for item in result["predictions"] if item["confidence"] >= threshold]
|
275 |
|
276 |
print(filtered_labels)
|
277 |
labels = filtered_labels
|
278 |
print(labels)
|
279 |
for class_ in labels:
|
280 |
+
total_cost += repair_costs.get(class_["class"],
|
281 |
+
0) * class_["confidence"]
|
282 |
result = {"total_cost": total_cost}
|
283 |
print(result)
|
284 |
|