jonathanjordan21 commited on
Commit
4242f3c
·
verified ·
1 Parent(s): 86dae20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -117,10 +117,12 @@ def respond(
117
  if is_multiple:
118
  request_details = []
119
  request_numbers = []
 
120
  for i,score in enumerate(scores):
121
  if score > threshold:
122
  request_details.append(codes[i][6:])
123
  request_numbers.append(codes[i][:3])
 
124
 
125
  if not request_details:
126
  request_details.append(undetected[6:])
@@ -128,8 +130,9 @@ def respond(
128
 
129
  request_numbers = "\n".join(request_numbers)
130
  request_details = "\n".join(request_details)
 
131
 
132
- return "Request code number:\n" + request_numbers + "\n\nRequest detail:\n" + request_details + "\n\nPlate numbers: " + plate_numbers
133
 
134
 
135
  s_max = scores.argmax()
@@ -140,7 +143,7 @@ def respond(
140
  else:
141
  request_code = codes[scores.argmax()]
142
 
143
- return "Request code number: " + request_code[:3] + "\nRequest detail: " + request_code[6:] + "\nPlate numbers: " + plate_numbers
144
 
145
  # for val in history:
146
  # if val[0]:
 
117
  if is_multiple:
118
  request_details = []
119
  request_numbers = []
120
+ request_scores = []
121
  for i,score in enumerate(scores):
122
  if score > threshold:
123
  request_details.append(codes[i][6:])
124
  request_numbers.append(codes[i][:3])
125
+ request_scores.append(score)
126
 
127
  if not request_details:
128
  request_details.append(undetected[6:])
 
130
 
131
  request_numbers = "\n".join(request_numbers)
132
  request_details = "\n".join(request_details)
133
+ request_scores = "\n".join(request_scores)
134
 
135
+ return "Request code number:\n" + request_numbers + "\n\nRequest detail:\n" + request_details + f"\nConfidence score:\n{request_scores}" + "\n\nPlate numbers: " + plate_numbers
136
 
137
 
138
  s_max = scores.argmax()
 
143
  else:
144
  request_code = codes[scores.argmax()]
145
 
146
+ return "Request code number: " + request_code[:3] + "\nRequest detail: " + request_code[6:] + f"\nConfidence score: {scores[s_max]}" + "\nPlate numbers: " + plate_numbers
147
 
148
  # for val in history:
149
  # if val[0]: