Spaces:
Runtime error
Runtime error
sshi
commited on
Commit
·
cbbced1
1
Parent(s):
063786d
”Add more description“
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ import cv2
|
|
| 7 |
import numpy
|
| 8 |
from transformers import AutoFeatureExtractor, AutoModelForObjectDetection
|
| 9 |
from PIL import Image
|
|
|
|
| 10 |
|
| 11 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 12 |
|
|
@@ -141,7 +142,11 @@ def detect(img):
|
|
| 141 |
return visualize_preds(img, preds)
|
| 142 |
|
| 143 |
|
| 144 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|
|
|
|
| 7 |
import numpy
|
| 8 |
from transformers import AutoFeatureExtractor, AutoModelForObjectDetection
|
| 9 |
from PIL import Image
|
| 10 |
+
import streamlit as st
|
| 11 |
|
| 12 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 13 |
|
|
|
|
| 142 |
return visualize_preds(img, preds)
|
| 143 |
|
| 144 |
|
| 145 |
+
st.markdown("Welcome to this space! 🤗")
|
| 146 |
+
|
| 147 |
+
|
| 148 |
+
|
| 149 |
+
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\n" + \
|
| 150 |
"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."
|
| 151 |
|
| 152 |
|