Hugging Face's logo

Spaces:
PierreSHI
/
YOLOS_traffic_object_detection
Runtime error

App Files Community
sshi commited on
Commit
27158a8
·
1 Parent(s): d252b7e

App bug fix.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -83,7 +83,7 @@ model_yolos.to(device)
83
  model_yolos.eval()
84
 
85
  # colors for visualization
86
- COLORS = [
87
  [0.000, 0.447, 0.741],
88
  [0.850, 0.325, 0.098],
89
  [0.929, 0.694, 0.125],
@@ -92,7 +92,7 @@ COLORS = [
92
  [0.301, 0.745, 0.933],
93
  [0.184, 0.494, 0.741],
94
  [0.494, 0.674, 0.556],
95
- [0.494, 0.301, 0.933],
96
  [0.000, 0.325, 0.850],
97
  [0.745, 0.301, 0.188]]
98
 
 
83
  model_yolos.eval()
84
 
85
  # colors for visualization
86
+ colors = [
87
  [0.000, 0.447, 0.741],
88
  [0.850, 0.325, 0.098],
89
  [0.929, 0.694, 0.125],
 
92
  [0.301, 0.745, 0.933],
93
  [0.184, 0.494, 0.741],
94
  [0.494, 0.674, 0.556],
95
+ [0.494, 0.301, 0.933],
96
  [0.000, 0.325, 0.850],
97
  [0.745, 0.301, 0.188]]
98