leedoming commited on
Commit
9184bfc
·
verified ·
1 Parent(s): bc3fea2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def load_yolos_model():
32
  yolos_processor, yolos_model = load_yolos_model()
33
 
34
  # Define the categories
35
- CATS = ['shirt, blouse', 'top, t-shirt, sweatshirt', 'sweater', 'cardigan', 'jacket', 'vest', 'pants', 'shorts', 'skirt', 'coat', 'dress', 'jumpsuit', 'cape', 'glasses', 'hat', 'headband, head covering, hair accessory', 'tie', 'glove', 'watch', 'belt', 'leg warmer', 'tights, stockings', 'sock', 'shoe', 'bag, wallet', 'scarf', 'umbrella', 'hood', 'collar', 'lapel', 'epaulette', 'sleeve', 'pocket', 'neckline', 'buckle', 'zipper', 'applique', 'bead', 'bow', 'flower', 'fringe', 'ribbon', 'rivet', 'ruffle', 'sequin', 'tassel']
36
 
37
  # Helper functions
38
  def load_image_from_url(url, max_retries=3):
 
32
  yolos_processor, yolos_model = load_yolos_model()
33
 
34
  # Define the categories
35
+ CATS = ['shirt, blouse', 'top, t-shirt, sweatshirt', 'sweater', 'cardigan', 'jacket', 'vest', 'pants', 'shorts', 'skirt', 'coat', 'dress', 'jumpsuit', 'cape', 'glasses', 'hat', 'headband, head covering, hair accessory', 'tie', 'glove', 'watch', 'belt', 'leg warmer', 'shoe', 'bag', 'wallet', 'scarf', 'umbrella', 'hood', 'collar', 'lapel', 'epaulette', 'buckle', 'applique', 'bead', 'bow', 'flower', 'fringe', 'ribbon', 'rivet', 'ruffle', 'sequin', 'tassel']
36
 
37
  # Helper functions
38
  def load_image_from_url(url, max_retries=3):