Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import pandas as pd
|
|
| 10 |
import time
|
| 11 |
from PIL import Image
|
| 12 |
|
| 13 |
-
TITLE = "Video Tagger"
|
| 14 |
DESCRIPTION = """
|
| 15 |
Upload a .mp4 or .mov video, choose how often to sample frames, and generate
|
| 16 |
combined (deduplicated) tags using a selected **tagging/captioning model**.
|
|
@@ -19,6 +19,7 @@ combined (deduplicated) tags using a selected **tagging/captioning model**.
|
|
| 19 |
- Control thresholds for **General Tags** and **Character Tags**.
|
| 20 |
- All tags from all sampled frames are merged into **one unique, comma-separated string**.
|
| 21 |
- Use the **Tag Control** tab to define tag substitutions and exclusions for the final output.
|
|
|
|
| 22 |
"""
|
| 23 |
|
| 24 |
DEFAULT_MODEL_REPO = "SmilingWolf/wd-eva02-large-tagger-v3"
|
|
@@ -457,7 +458,7 @@ class VideoTagger:
|
|
| 457 |
pct,
|
| 458 |
desc=(
|
| 459 |
f"Preparing batch {current_batch}/{total_batches} "
|
| 460 |
-
f"({len(batch_tensors)}/{current_batch_size} frames)
|
| 461 |
),
|
| 462 |
)
|
| 463 |
|
|
@@ -471,7 +472,7 @@ class VideoTagger:
|
|
| 471 |
pct,
|
| 472 |
desc=(
|
| 473 |
f"Processing batch {current_batch}/{total_batches} "
|
| 474 |
-
f"(frames {beg}-{end}/{sampled_frames})
|
| 475 |
),
|
| 476 |
)
|
| 477 |
|
|
@@ -513,7 +514,7 @@ class VideoTagger:
|
|
| 513 |
pct,
|
| 514 |
desc=(
|
| 515 |
f"Processing final batch {current_batch}/{total_batches} "
|
| 516 |
-
f"(frames {beg}-{end}/{sampled_frames})
|
| 517 |
),
|
| 518 |
)
|
| 519 |
|
|
|
|
| 10 |
import time
|
| 11 |
from PIL import Image
|
| 12 |
|
| 13 |
+
TITLE = "AI Video Auto-Tagger & Captioner"
|
| 14 |
DESCRIPTION = """
|
| 15 |
Upload a .mp4 or .mov video, choose how often to sample frames, and generate
|
| 16 |
combined (deduplicated) tags using a selected **tagging/captioning model**.
|
|
|
|
| 19 |
- Control thresholds for **General Tags** and **Character Tags**.
|
| 20 |
- All tags from all sampled frames are merged into **one unique, comma-separated string**.
|
| 21 |
- Use the **Tag Control** tab to define tag substitutions and exclusions for the final output.
|
| 22 |
+
**This space is running on the free CPU tier so it can be slow. If you want better speeds, clone the space and host it on more capable hardware.**
|
| 23 |
"""
|
| 24 |
|
| 25 |
DEFAULT_MODEL_REPO = "SmilingWolf/wd-eva02-large-tagger-v3"
|
|
|
|
| 458 |
pct,
|
| 459 |
desc=(
|
| 460 |
f"Preparing batch {current_batch}/{total_batches} "
|
| 461 |
+
f"({len(batch_tensors)}/{current_batch_size} frames)"
|
| 462 |
),
|
| 463 |
)
|
| 464 |
|
|
|
|
| 472 |
pct,
|
| 473 |
desc=(
|
| 474 |
f"Processing batch {current_batch}/{total_batches} "
|
| 475 |
+
f"(frames {beg}-{end}/{sampled_frames})"
|
| 476 |
),
|
| 477 |
)
|
| 478 |
|
|
|
|
| 514 |
pct,
|
| 515 |
desc=(
|
| 516 |
f"Processing final batch {current_batch}/{total_batches} "
|
| 517 |
+
f"(frames {beg}-{end}/{sampled_frames})"
|
| 518 |
),
|
| 519 |
)
|
| 520 |
|