Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def tag_image(pil_image: Image.Image, output_format: str) -> str:
|
|
32 |
idx_to_tag = metadata["idx_to_tag"]
|
33 |
tag_to_category = metadata.get("tag_to_category", {})
|
34 |
category_thresholds = metadata.get("category_thresholds", {})
|
35 |
-
default_threshold = 0.
|
36 |
results_by_cat = {} # to store tags per category (for verbose output)
|
37 |
prompt_tags = [] # to store tags for prompt-style output
|
38 |
# Collect tags above thresholds
|
|
|
32 |
idx_to_tag = metadata["idx_to_tag"]
|
33 |
tag_to_category = metadata.get("tag_to_category", {})
|
34 |
category_thresholds = metadata.get("category_thresholds", {})
|
35 |
+
default_threshold = 0.35
|
36 |
results_by_cat = {} # to store tags per category (for verbose output)
|
37 |
prompt_tags = [] # to store tags for prompt-style output
|
38 |
# Collect tags above thresholds
|