yym68686 commited on
Commit
ed95244
·
1 Parent(s): 5eb10aa

🐛 Bug: Fix the bug of middleware statistics for unknown models

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -127,7 +127,8 @@ class StatsMiddleware(BaseHTTPMiddleware):
127
  self.request_times[endpoint] += process_time
128
  self.ip_counts[endpoint][client_ip] += 1
129
  self.request_arrivals[endpoint].append(arrival_time)
130
- self.model_counts[model] += 1
 
131
 
132
  return response
133
 
 
127
  self.request_times[endpoint] += process_time
128
  self.ip_counts[endpoint][client_ip] += 1
129
  self.request_arrivals[endpoint].append(arrival_time)
130
+ if model != "unknown":
131
+ self.model_counts[model] += 1
132
 
133
  return response
134