Spaces:
Running
on
Zero
Running
on
Zero
Upload app.py
Browse files
app.py
CHANGED
|
@@ -164,7 +164,6 @@ def get_tags(probs, labels: LabelData, gen_threshold, char_threshold):
|
|
| 164 |
"copyright": (labels.copyright, char_threshold),
|
| 165 |
"artist": (labels.artist, char_threshold),
|
| 166 |
"meta": (labels.meta, gen_threshold),
|
| 167 |
-
"quality": (labels.quality, gen_threshold),
|
| 168 |
"model": (labels.model, gen_threshold)
|
| 169 |
}
|
| 170 |
for category, (indices, threshold) in category_map.items():
|
|
|
|
| 164 |
"copyright": (labels.copyright, char_threshold),
|
| 165 |
"artist": (labels.artist, char_threshold),
|
| 166 |
"meta": (labels.meta, gen_threshold),
|
|
|
|
| 167 |
"model": (labels.model, gen_threshold)
|
| 168 |
}
|
| 169 |
for category, (indices, threshold) in category_map.items():
|