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

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', '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):
 
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', 'glove', 'shoe', 'bag', 'wallet', 'umbrella', 'hood', 'collar', 'lapel', 'epaulette', 'applique', 'bead', 'bow', 'flower', 'fringe', 'ribbon', 'rivet', 'ruffle', 'sequin', 'tassel']
36
 
37
  # Helper functions
38
  def load_image_from_url(url, max_retries=3):