throaway2854 commited on
Commit
1bc957a
·
verified ·
1 Parent(s): 364fd6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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