sshi commited on
Commit
063786d
1 Parent(s): 8004ebd

Modify description..

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -19,10 +19,10 @@ colors = [
19
  [ 0, 113, 188,],
20
  [216, 82, 24,],
21
  [236, 176, 31,],
22
- [192, 202, 25,],
23
  [118, 171, 47,],
24
  [ 76, 189, 237,],
25
- [ 46, 125, 188,],
26
  [125, 171, 141,],
27
  [125, 76, 237,],
28
  [ 0, 82, 216,],
@@ -141,8 +141,8 @@ def detect(img):
141
  return visualize_preds(img, preds)
142
 
143
 
144
- description = "This is a traffic object detector based on <a href='https://huggingface.co/docs/transformers/model_doc/yolos' style='text-decoration: underline' target='_blank'>YOLOS</a>. \n" + \
145
- "The model can detect following targets: {1: 'person', 2: 'rider', 3: 'car', 4: 'bus', 5: 'truck', 6: 'bike', 7: 'motor', 8: 'traffic light', 9: 'traffic sign', 10: 'train'}."
146
 
147
 
148
  interface = gr.Interface(
 
19
  [ 0, 113, 188,],
20
  [216, 82, 24,],
21
  [236, 176, 31,],
22
+ [255, 255, 0,],
23
  [118, 171, 47,],
24
  [ 76, 189, 237,],
25
+ [ 46, 155, 188,],
26
  [125, 171, 141,],
27
  [125, 76, 237,],
28
  [ 0, 82, 216,],
 
141
  return visualize_preds(img, preds)
142
 
143
 
144
+ description = "this is a traffic object detector based on <a href='https://huggingface.co/docs/transformers/model_doc/yolos' style='text-decoration: underline' target='_blank'>YOLOS</a>. \n" + \
145
+ "The model can detect following targets: 1 person, 2 rider, 3 car, 4 bus, 5 truck, 6 bike, 7 motor, 8 traffic light, 9 traffic sign, 10 train."
146
 
147
 
148
  interface = gr.Interface(