sshi commited on
Commit
60390c6
1 Parent(s): cbbced1

Add emoji in the description.

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -139,15 +139,11 @@ def visualize_preds(image, preds, threshold=0.9):
139
  def detect(img):
140
  # Run inference
141
  preds = generate_preds(feature_extractor, model_yolos, img)
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
 
153
  interface = gr.Interface(
 
139
  def detect(img):
140
  # Run inference
141
  preds = generate_preds(feature_extractor, model_yolos, img)
142
+ return visualize_preds(img, preds)
143
 
144
 
145
+ description = "Welcome to this space! 🤗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" + \
146
+ "The model can detect following targets: person🚶‍♂️, rider🚴‍♀️, car🚗, bus🚌, truck🚚, bike🚲, motor🏍️, traffic light🚦, traffic sign⛔, train🚄."
 
 
 
 
147
 
148
 
149
  interface = gr.Interface(