Update app.py
Browse files
app.py
CHANGED
|
@@ -30,6 +30,12 @@ MODEL_OPTIONS = [
|
|
| 30 |
"SmilingWolf/wd-swinv2-tagger-v3",
|
| 31 |
"deepghs/idolsankaku-eva02-large-tagger-v1",
|
| 32 |
"deepghs/idolsankaku-swinv2-tagger-v1",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
]
|
| 34 |
|
| 35 |
MODEL_FILENAME = "model.onnx"
|
|
@@ -510,7 +516,7 @@ with gr.Blocks(title=TITLE) as demo:
|
|
| 510 |
tag_substitutes_df = gr.Dataframe(
|
| 511 |
headers=["Original Tag", "Substitute Tag"],
|
| 512 |
datatype=["str", "str"],
|
| 513 |
-
row_count=(
|
| 514 |
col_count=(2, "fixed"),
|
| 515 |
type="array",
|
| 516 |
label="Tag Substitutes",
|
|
@@ -525,7 +531,7 @@ with gr.Blocks(title=TITLE) as demo:
|
|
| 525 |
tag_exclusions_df = gr.Dataframe(
|
| 526 |
headers=["Tag to Exclude"],
|
| 527 |
datatype=["str"],
|
| 528 |
-
row_count=(
|
| 529 |
col_count=(1, "fixed"),
|
| 530 |
type="array",
|
| 531 |
label="Tag Exclusions",
|
|
|
|
| 30 |
"SmilingWolf/wd-swinv2-tagger-v3",
|
| 31 |
"deepghs/idolsankaku-eva02-large-tagger-v1",
|
| 32 |
"deepghs/idolsankaku-swinv2-tagger-v1",
|
| 33 |
+
"gokaygokay/Florence-2-SD3-Captioner",
|
| 34 |
+
"gokaygokay/Florence-2-Flux",
|
| 35 |
+
"gokaygokay/Florence-2-Flux-Large",
|
| 36 |
+
"MiaoshouAI/Florence-2-large-PromptGen-v2.0",
|
| 37 |
+
"thwri/CogFlorence-2.2-Large",
|
| 38 |
+
"deepghs/deepgelbooru_onnx",
|
| 39 |
]
|
| 40 |
|
| 41 |
MODEL_FILENAME = "model.onnx"
|
|
|
|
| 516 |
tag_substitutes_df = gr.Dataframe(
|
| 517 |
headers=["Original Tag", "Substitute Tag"],
|
| 518 |
datatype=["str", "str"],
|
| 519 |
+
row_count=(20, "dynamic"),
|
| 520 |
col_count=(2, "fixed"),
|
| 521 |
type="array",
|
| 522 |
label="Tag Substitutes",
|
|
|
|
| 531 |
tag_exclusions_df = gr.Dataframe(
|
| 532 |
headers=["Tag to Exclude"],
|
| 533 |
datatype=["str"],
|
| 534 |
+
row_count=(20, "dynamic"),
|
| 535 |
col_count=(1, "fixed"),
|
| 536 |
type="array",
|
| 537 |
label="Tag Exclusions",
|