reidddd commited on
Commit
a129ece
·
1 Parent(s): b0086c4

updated roboflow model to improve accuracy

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -233,7 +233,7 @@ def fetchImage():
233
  result = CLIENT.infer(
234
  file_name, model_id="car-recognition-v4/4")
235
  filtered_labels = [item
236
- for item in result["predictions"] if item["confidence"] >= threshold]
237
 
238
  if filtered_labels == []:
239
  i = 0.7
@@ -244,9 +244,8 @@ def fetchImage():
244
  with CLIENT.use_configuration(custom_configuration):
245
  result = CLIENT.infer(
246
  file_name, model_id="car-recognition-v4/4")
247
- threshold = 0.3 # Adjust this as needed
248
  filtered_labels = [item
249
- for item in result["predictions"] if item["confidence"] >= threshold]
250
  labels = filtered_labels
251
  print(labels)
252
  for class_ in labels:
 
233
  result = CLIENT.infer(
234
  file_name, model_id="car-recognition-v4/4")
235
  filtered_labels = [item
236
+ for item in result["predictions"]]
237
 
238
  if filtered_labels == []:
239
  i = 0.7
 
244
  with CLIENT.use_configuration(custom_configuration):
245
  result = CLIENT.infer(
246
  file_name, model_id="car-recognition-v4/4")
 
247
  filtered_labels = [item
248
+ for item in result["predictions"]]
249
  labels = filtered_labels
250
  print(labels)
251
  for class_ in labels: