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

”Add more description“

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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
- 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
 
 
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