Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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', '
|
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):
|