Spaces:
Runtime error
Runtime error
sshi
commited on
Commit
·
063786d
1
Parent(s):
8004ebd
Modify description..
Browse files
app.py
CHANGED
|
@@ -19,10 +19,10 @@ colors = [
|
|
| 19 |
[ 0, 113, 188,],
|
| 20 |
[216, 82, 24,],
|
| 21 |
[236, 176, 31,],
|
| 22 |
-
[
|
| 23 |
[118, 171, 47,],
|
| 24 |
[ 76, 189, 237,],
|
| 25 |
-
[ 46,
|
| 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 = "
|
| 145 |
-
"The model can detect following targets:
|
| 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(
|