reidddd commited on
Commit
167361f
·
1 Parent(s): 5c60806

updated roboflow model to improve accuracy

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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["class"]
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_, 0)
 
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