Gordon Li commited on
Commit
c6ee923
·
1 Parent(s): 7877659
Files changed (1) hide show
  1. TrafficSpot.py +1 -2
TrafficSpot.py CHANGED
@@ -94,7 +94,6 @@ class TrafficSpot:
94
  <div style='border-top: 1px solid #ccc; padding: 5px 0;'>
95
  <p style='margin: 2px 0;'><strong>Time:</strong> {escape(str(row['capture_time']))}</p>
96
  <p style='margin: 2px 0;'><strong>Vehicles:</strong> {escape(str(row['vehicle_count']))}</p>
97
- <p style='margin: 2px 0;'><strong>Confidence:</strong> {escape(str(row['confidence_score']))}</p>
98
  {image_html}
99
  </div>
100
  """
@@ -198,7 +197,7 @@ class TrafficSpotManager:
198
  return
199
 
200
  try:
201
- dataset = load_dataset("slliac/traffic-analysis", split="train")
202
  dataset_list = [row for row in dataset]
203
  dataset_list.sort(key=lambda x: x['capture_time'], reverse=True)
204
 
 
94
  <div style='border-top: 1px solid #ccc; padding: 5px 0;'>
95
  <p style='margin: 2px 0;'><strong>Time:</strong> {escape(str(row['capture_time']))}</p>
96
  <p style='margin: 2px 0;'><strong>Vehicles:</strong> {escape(str(row['vehicle_count']))}</p>
 
97
  {image_html}
98
  </div>
99
  """
 
197
  return
198
 
199
  try:
200
+ dataset = load_dataset("slliac/isom5240-td-application-traffic-analysis", split="application")
201
  dataset_list = [row for row in dataset]
202
  dataset_list.sort(key=lambda x: x['capture_time'], reverse=True)
203