Spaces:
Running
on
Zero
Running
on
Zero
Upload tagger.py
Browse files
tagger.py
CHANGED
@@ -436,7 +436,7 @@ def predict_tags(image: Image.Image, general_threshold: float = 0.3, character_t
|
|
436 |
|
437 |
|
438 |
def predict_tags_wd(image: Image.Image, input_tags: str, algo: list[str], general_threshold: float = 0.3, character_threshold: float = 0.8):
|
439 |
-
if
|
440 |
return "", "", input_tags, gr.update(interactive=True),
|
441 |
return predict_tags(image, general_threshold, character_threshold)
|
442 |
|
|
|
436 |
|
437 |
|
438 |
def predict_tags_wd(image: Image.Image, input_tags: str, algo: list[str], general_threshold: float = 0.3, character_threshold: float = 0.8):
|
439 |
+
if not "Use WD Tagger" in algo and len(algo) != 0:
|
440 |
return "", "", input_tags, gr.update(interactive=True),
|
441 |
return predict_tags(image, general_threshold, character_threshold)
|
442 |
|