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 |
|