diff --git a/Cargo.lock b/Cargo.lock
index fcc0e62..3a44f57 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -453,8 +453,8 @@ checksum = "8318a53db07bb3f8dca91a600466bdb3f2eaadeedfdbcf02e1accbad9271ba50"
 
 [[package]]
 name = "candle-core"
-version = "0.7.2"
-source = "git+https://git.v0l.io/huggingface/candle.git?tag=0.7.2#3a3c48b14b861da0d275cae44edb1b0191f83bf4"
+version = "0.8.1"
+source = "git+https://git.v0l.io/huggingface/candle.git?tag=0.8.1#67cab7d6b8279f953b0a8cc5012b135b9743cdc8"
 dependencies = [
  "byteorder",
  "gemm",
@@ -467,14 +467,15 @@ dependencies = [
  "rayon",
  "safetensors",
  "thiserror",
+ "ug",
  "yoke",
  "zip",
 ]
 
 [[package]]
 name = "candle-nn"
-version = "0.7.2"
-source = "git+https://git.v0l.io/huggingface/candle.git?tag=0.7.2#3a3c48b14b861da0d275cae44edb1b0191f83bf4"
+version = "0.8.1"
+source = "git+https://git.v0l.io/huggingface/candle.git?tag=0.8.1#67cab7d6b8279f953b0a8cc5012b135b9743cdc8"
 dependencies = [
  "candle-core",
  "half",
@@ -487,8 +488,8 @@ dependencies = [
 
 [[package]]
 name = "candle-transformers"
-version = "0.7.2"
-source = "git+https://git.v0l.io/huggingface/candle.git?tag=0.7.2#3a3c48b14b861da0d275cae44edb1b0191f83bf4"
+version = "0.8.1"
+source = "git+https://git.v0l.io/huggingface/candle.git?tag=0.8.1#67cab7d6b8279f953b0a8cc5012b135b9743cdc8"
 dependencies = [
  "byteorder",
  "candle-core",
@@ -2149,6 +2150,30 @@ dependencies = [
  "winapi",
 ]
 
+[[package]]
+name = "num"
+version = "0.4.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "35bd024e8b2ff75562e5f34e7f4905839deb4b22955ef5e73d2fea1b9813cb23"
+dependencies = [
+ "num-bigint",
+ "num-complex",
+ "num-integer",
+ "num-iter",
+ "num-rational",
+ "num-traits",
+]
+
+[[package]]
+name = "num-bigint"
+version = "0.4.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a5e44f723f1133c9deac646763579fdb3ac745e418f2a7af9cd0c431da1f20b9"
+dependencies = [
+ "num-integer",
+ "num-traits",
+]
+
 [[package]]
 name = "num-bigint-dig"
 version = "0.8.4"
@@ -2202,6 +2227,17 @@ dependencies = [
  "num-traits",
 ]
 
+[[package]]
+name = "num-rational"
+version = "0.4.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f83d14da390562dca69fc84082e73e548e1ad308d24accdedd2720017cb37824"
+dependencies = [
+ "num-bigint",
+ "num-integer",
+ "num-traits",
+]
+
 [[package]]
 name = "num-traits"
 version = "0.2.19"
@@ -4104,6 +4140,19 @@ version = "0.1.6"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "ed646292ffc8188ef8ea4d1e0e0150fb15a5c2e12ad9b8fc191ae7a8a7f3c4b9"
 
+[[package]]
+name = "ug"
+version = "0.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c4eef2ebfc18c67a6dbcacd9d8a4d85e0568cc58c82515552382312c2730ea13"
+dependencies = [
+ "half",
+ "num",
+ "serde",
+ "serde_json",
+ "thiserror",
+]
+
 [[package]]
 name = "uncased"
 version = "0.9.10"
diff --git a/Cargo.toml b/Cargo.toml
index b74e156..8f3868b 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -47,8 +47,8 @@ clap = { version = "4.5.18", features = ["derive"] }
 
 libc = { version = "0.2.153", optional = true }
 ffmpeg-rs-raw = { git = "https://git.v0l.io/Kieran/ffmpeg-rs-raw.git", rev = "df69b2f05da4279e36ad55086d77b45b2caf5174", optional = true }
-candle-core = { git = "https://git.v0l.io/huggingface/candle.git", tag = "0.7.2", optional = true }
-candle-nn = { git = "https://git.v0l.io/huggingface/candle.git", tag = "0.7.2", optional = true }
-candle-transformers = { git = "https://git.v0l.io/huggingface/candle.git", tag = "0.7.2", optional = true }
+candle-core = { git = "https://git.v0l.io/huggingface/candle.git", tag = "0.8.1", optional = true }
+candle-nn = { git = "https://git.v0l.io/huggingface/candle.git", tag = "0.8.1", optional = true }
+candle-transformers = { git = "https://git.v0l.io/huggingface/candle.git", tag = "0.8.1", optional = true }
 sqlx-postgres = { version = "0.8.2", optional = true, features = ["chrono", "uuid"] }
 mime2ext = "0.1.53"
diff --git a/config.toml b/config.toml
index d6ae75d..84187fc 100644
--- a/config.toml
+++ b/config.toml
@@ -17,7 +17,10 @@ public_url = "http://localhost:8000"
 # whitelist = ["63fe6318dc58583cfe16810f86dd09e18bfd76aabc24a0081ce2856f330504ed"]
 
 # Path for ViT(224) image model (https://huggingface.co/google/vit-base-patch16-224)
-# vit_model_path = "model.safetennsors"
+#vit_model_path = "falcon_nsfw.safetensors"
+
+# Path for ViT(224) image model config.json
+#vit_model_config_path = "falcon_nsfw.json"
 
 # Webhook api endpoint
 # webhook_url = "https://api.snort.social/api/v1/media/webhook"
diff --git a/src/filesystem.rs b/src/filesystem.rs
index 99401d7..b9cca12 100644
--- a/src/filesystem.rs
+++ b/src/filesystem.rs
@@ -109,9 +109,14 @@ impl FileStore {
 
                 #[cfg(feature = "labels")]
                 let labels = if let Some(mp) = &self.settings.vit_model_path {
-                    label_frame(&new_temp.result, mp.clone())?
+                    let config_path = if let Some(c) = &self.settings.vit_model_config_path {
+                        c
+                    } else {
+                        bail!("Missing vit_model_config_path");
+                    };
+                    label_frame(&new_temp.result, mp.clone(), config_path.clone())?
                         .iter()
-                        .map(|l| FileLabel::new(l.clone(), "vit224".to_string()))
+                        .map(|l| FileLabel::new(l.0.clone(), "vit224".to_string()))
                         .collect()
                 } else {
                     vec![]
diff --git a/src/processing/labeling.rs b/src/processing/labeling.rs
index b3a7300..6d17504 100644
--- a/src/processing/labeling.rs
+++ b/src/processing/labeling.rs
@@ -1,3 +1,4 @@
+use std::collections::HashMap;
 use std::path::{Path, PathBuf};
 use std::slice;
 
@@ -8,14 +9,24 @@ use candle_transformers::models::vit;
 use ffmpeg_rs_raw::ffmpeg_sys_the_third::AVPixelFormat::AV_PIX_FMT_RGB24;
 use ffmpeg_rs_raw::ffmpeg_sys_the_third::{av_frame_free, av_packet_free};
 use ffmpeg_rs_raw::{Decoder, Demuxer, Scaler};
+use nostr::serde_json;
+use serde::Deserialize;
 
-pub fn label_frame(frame: &Path, model: PathBuf) -> Result<Vec<String>> {
+#[derive(Deserialize)]
+struct MyVitConfig {
+    pub id2label: HashMap<usize, String>,
+}
+
+pub fn label_frame(frame: &Path, model: PathBuf, config: PathBuf) -> Result<HashMap<String, f32>> {
     unsafe {
         let device = Device::Cpu;
         let image = load_frame_224(frame)?.to_device(&device)?;
 
         let vb = VarBuilder::from_mmaped_safetensors(&[model], DType::F32, &device)?;
-        let model = vit::Model::new(&vit::Config::vit_base_patch16_224(), 1000, vb)?;
+        let config_data = std::fs::read(config)?;
+        let config: vit::Config = serde_json::from_slice(&config_data)?;
+        let label_config: MyVitConfig = serde_json::from_slice(&config_data)?;
+        let model = vit::Model::new(&config, label_config.id2label.len(), vb)?;
         let logits = model.forward(&image.unsqueeze(0)?)?;
         let prs = candle_nn::ops::softmax(&logits, D::Minus1)?
             .i(0)?
@@ -24,10 +35,11 @@ pub fn label_frame(frame: &Path, model: PathBuf) -> Result<Vec<String>> {
         prs.sort_by(|(_, p1), (_, p2)| p2.total_cmp(p1));
         let res = prs
             .iter()
-            //.filter(|&(_c, q)| **q >= 0.50f32)
+            //.filter(|&(_c, q)| **q >= 0.1)
             .take(5)
-            .map(|&(c, _q)| CLASSES[c].to_string())
+            .map(|&(c, q)| (label_config.id2label[&c].to_string(), *q))
             .collect();
+        println!("prs: {:?}", res);
         Ok(res)
     }
 }
@@ -54,7 +66,7 @@ unsafe fn load_image(path_buf: &Path, width: usize, height: usize) -> Result<Vec
                 let line_size = (*new_frame).linesize[0] as usize;
                 let row_offset = line_size * row;
                 let row_slice =
-                    slice::from_raw_parts((*new_frame).data[0].add(row_offset), line_size);
+                    slice::from_raw_parts((*new_frame).data[0].add(row_offset), 3 * width);
                 dst_vec.extend_from_slice(row_slice);
             }
             av_frame_free(&mut frame);
@@ -66,1018 +78,17 @@ unsafe fn load_image(path_buf: &Path, width: usize, height: usize) -> Result<Vec
     Err(Error::msg("No image data found"))
 }
 
+// https://github.com/huggingface/candle/blob/main/candle-examples/src/imagenet.rs
 unsafe fn load_frame_224(path: &Path) -> Result<Tensor> {
     let pic = load_image(path, 224, 224)?;
 
-    //fs::write("frame_224.raw", &pic_slice)?;
-    let data = Tensor::from_vec(pic, (224, 224, 3), &Device::Cpu)?.permute((2, 0, 1))?;
-    let mean = Tensor::new(&[0.485f32, 0.456, 0.406], &Device::Cpu)?.reshape((3, 1, 1))?;
-    let std = Tensor::new(&[0.229f32, 0.224, 0.225], &Device::Cpu)?.reshape((3, 1, 1))?;
+    std::fs::write("frame_224.raw", &pic)?;
+    let d = Device::cuda_if_available(0)?;
+    let data = Tensor::from_vec(pic, (224, 224, 3), &d)?.permute((2, 0, 1))?;
+    let mean = Tensor::new(&[0.485f32, 0.456, 0.406], &d)?.reshape((3, 1, 1))?;
+    let std = Tensor::new(&[0.229f32, 0.224, 0.225], &d)?.reshape((3, 1, 1))?;
     let res = (data.to_dtype(DType::F32)? / 255.)?
         .broadcast_sub(&mean)?
         .broadcast_div(&std)?;
     Ok(res)
 }
-
-pub const CLASSES: [&str; 1000] = [
-    "tench, Tinca tinca",
-    "goldfish, Carassius auratus",
-    "great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias",
-    "tiger shark, Galeocerdo cuvieri",
-    "hammerhead, hammerhead shark",
-    "electric ray, crampfish, numbfish, torpedo",
-    "stingray",
-    "cock",
-    "hen",
-    "ostrich, Struthio camelus",
-    "brambling, Fringilla montifringilla",
-    "goldfinch, Carduelis carduelis",
-    "house finch, linnet, Carpodacus mexicanus",
-    "junco, snowbird",
-    "indigo bunting, indigo finch, indigo bird, Passerina cyanea",
-    "robin, American robin, Turdus migratorius",
-    "bulbul",
-    "jay",
-    "magpie",
-    "chickadee",
-    "water ouzel, dipper",
-    "kite",
-    "bald eagle, American eagle, Haliaeetus leucocephalus",
-    "vulture",
-    "great grey owl, great gray owl, Strix nebulosa",
-    "European fire salamander, Salamandra salamandra",
-    "common newt, Triturus vulgaris",
-    "eft",
-    "spotted salamander, Ambystoma maculatum",
-    "axolotl, mud puppy, Ambystoma mexicanum",
-    "bullfrog, Rana catesbeiana",
-    "tree frog, tree-frog",
-    "tailed frog, bell toad, ribbed toad, tailed toad, Ascaphus trui",
-    "loggerhead, loggerhead turtle, Caretta caretta",
-    "leatherback turtle, leatherback, leathery turtle, Dermochelys coriacea",
-    "mud turtle",
-    "terrapin",
-    "box turtle, box tortoise",
-    "banded gecko",
-    "common iguana, iguana, Iguana iguana",
-    "American chameleon, anole, Anolis carolinensis",
-    "whiptail, whiptail lizard",
-    "agama",
-    "frilled lizard, Chlamydosaurus kingi",
-    "alligator lizard",
-    "Gila monster, Heloderma suspectum",
-    "green lizard, Lacerta viridis",
-    "African chameleon, Chamaeleo chamaeleon",
-    "Komodo dragon, Komodo lizard, dragon lizard, giant lizard, Varanus komodoensis",
-    "African crocodile, Nile crocodile, Crocodylus niloticus",
-    "American alligator, Alligator mississipiensis",
-    "triceratops",
-    "thunder snake, worm snake, Carphophis amoenus",
-    "ringneck snake, ring-necked snake, ring snake",
-    "hognose snake, puff adder, sand viper",
-    "green snake, grass snake",
-    "king snake, kingsnake",
-    "garter snake, grass snake",
-    "water snake",
-    "vine snake",
-    "night snake, Hypsiglena torquata",
-    "boa constrictor, Constrictor constrictor",
-    "rock python, rock snake, Python sebae",
-    "Indian cobra, Naja naja",
-    "green mamba",
-    "sea snake",
-    "horned viper, cerastes, sand viper, horned asp, Cerastes cornutus",
-    "diamondback, diamondback rattlesnake, Crotalus adamanteus",
-    "sidewinder, horned rattlesnake, Crotalus cerastes",
-    "trilobite",
-    "harvestman, daddy longlegs, Phalangium opilio",
-    "scorpion",
-    "black and gold garden spider, Argiope aurantia",
-    "barn spider, Araneus cavaticus",
-    "garden spider, Aranea diademata",
-    "black widow, Latrodectus mactans",
-    "tarantula",
-    "wolf spider, hunting spider",
-    "tick",
-    "centipede",
-    "black grouse",
-    "ptarmigan",
-    "ruffed grouse, partridge, Bonasa umbellus",
-    "prairie chicken, prairie grouse, prairie fowl",
-    "peacock",
-    "quail",
-    "partridge",
-    "African grey, African gray, Psittacus erithacus",
-    "macaw",
-    "sulphur-crested cockatoo, Kakatoe galerita, Cacatua galerita",
-    "lorikeet",
-    "coucal",
-    "bee eater",
-    "hornbill",
-    "hummingbird",
-    "jacamar",
-    "toucan",
-    "drake",
-    "red-breasted merganser, Mergus serrator",
-    "goose",
-    "black swan, Cygnus atratus",
-    "tusker",
-    "echidna, spiny anteater, anteater",
-    "platypus, duckbill, duckbilled platypus, duck-billed platypus, Ornithorhynchus anatinus",
-    "wallaby, brush kangaroo",
-    "koala, koala bear, kangaroo bear, native bear, Phascolarctos cinereus",
-    "wombat",
-    "jellyfish",
-    "sea anemone, anemone",
-    "brain coral",
-    "flatworm, platyhelminth",
-    "nematode, nematode worm, roundworm",
-    "conch",
-    "snail",
-    "slug",
-    "sea slug, nudibranch",
-    "chiton, coat-of-mail shell, sea cradle, polyplacophore",
-    "chambered nautilus, pearly nautilus, nautilus",
-    "Dungeness crab, Cancer magister",
-    "rock crab, Cancer irroratus",
-    "fiddler crab",
-    "king crab, Alaska crab, Alaskan king crab, Alaska king crab, Paralithodes camtschatica",
-    "American lobster, Northern lobster, Maine lobster, Homarus americanus",
-    "spiny lobster, langouste, rock lobster, crawfish, crayfish, sea crawfish",
-    "crayfish, crawfish, crawdad, crawdaddy",
-    "hermit crab",
-    "isopod",
-    "white stork, Ciconia ciconia",
-    "black stork, Ciconia nigra",
-    "spoonbill",
-    "flamingo",
-    "little blue heron, Egretta caerulea",
-    "American egret, great white heron, Egretta albus",
-    "bittern",
-    "crane",
-    "limpkin, Aramus pictus",
-    "European gallinule, Porphyrio porphyrio",
-    "American coot, marsh hen, mud hen, water hen, Fulica americana",
-    "bustard",
-    "ruddy turnstone, Arenaria interpres",
-    "red-backed sandpiper, dunlin, Erolia alpina",
-    "redshank, Tringa totanus",
-    "dowitcher",
-    "oystercatcher, oyster catcher",
-    "pelican",
-    "king penguin, Aptenodytes patagonica",
-    "albatross, mollymawk",
-    "grey whale, gray whale, devilfish, Eschrichtius gibbosus, Eschrichtius robustus",
-    "killer whale, killer, orca, grampus, sea wolf, Orcinus orca",
-    "dugong, Dugong dugon",
-    "sea lion",
-    "Chihuahua",
-    "Japanese spaniel",
-    "Maltese dog, Maltese terrier, Maltese",
-    "Pekinese, Pekingese, Peke",
-    "Shih-Tzu",
-    "Blenheim spaniel",
-    "papillon",
-    "toy terrier",
-    "Rhodesian ridgeback",
-    "Afghan hound, Afghan",
-    "basset, basset hound",
-    "beagle",
-    "bloodhound, sleuthhound",
-    "bluetick",
-    "black-and-tan coonhound",
-    "Walker hound, Walker foxhound",
-    "English foxhound",
-    "redbone",
-    "borzoi, Russian wolfhound",
-    "Irish wolfhound",
-    "Italian greyhound",
-    "whippet",
-    "Ibizan hound, Ibizan Podenco",
-    "Norwegian elkhound, elkhound",
-    "otterhound, otter hound",
-    "Saluki, gazelle hound",
-    "Scottish deerhound, deerhound",
-    "Weimaraner",
-    "Staffordshire bullterrier, Staffordshire bull terrier",
-    "American Staffordshire terrier, Staffordshire terrier, American pit bull terrier, pit bull terrier",
-    "Bedlington terrier",
-    "Border terrier",
-    "Kerry blue terrier",
-    "Irish terrier",
-    "Norfolk terrier",
-    "Norwich terrier",
-    "Yorkshire terrier",
-    "wire-haired fox terrier",
-    "Lakeland terrier",
-    "Sealyham terrier, Sealyham",
-    "Airedale, Airedale terrier",
-    "cairn, cairn terrier",
-    "Australian terrier",
-    "Dandie Dinmont, Dandie Dinmont terrier",
-    "Boston bull, Boston terrier",
-    "miniature schnauzer",
-    "giant schnauzer",
-    "standard schnauzer",
-    "Scotch terrier, Scottish terrier, Scottie",
-    "Tibetan terrier, chrysanthemum dog",
-    "silky terrier, Sydney silky",
-    "soft-coated wheaten terrier",
-    "West Highland white terrier",
-    "Lhasa, Lhasa apso",
-    "flat-coated retriever",
-    "curly-coated retriever",
-    "golden retriever",
-    "Labrador retriever",
-    "Chesapeake Bay retriever",
-    "German short-haired pointer",
-    "vizsla, Hungarian pointer",
-    "English setter",
-    "Irish setter, red setter",
-    "Gordon setter",
-    "Brittany spaniel",
-    "clumber, clumber spaniel",
-    "English springer, English springer spaniel",
-    "Welsh springer spaniel",
-    "cocker spaniel, English cocker spaniel, cocker",
-    "Sussex spaniel",
-    "Irish water spaniel",
-    "kuvasz",
-    "schipperke",
-    "groenendael",
-    "malinois",
-    "briard",
-    "kelpie",
-    "komondor",
-    "Old English sheepdog, bobtail",
-    "Shetland sheepdog, Shetland sheep dog, Shetland",
-    "collie",
-    "Border collie",
-    "Bouvier des Flandres, Bouviers des Flandres",
-    "Rottweiler",
-    "German shepherd, German shepherd dog, German police dog, alsatian",
-    "Doberman, Doberman pinscher",
-    "miniature pinscher",
-    "Greater Swiss Mountain dog",
-    "Bernese mountain dog",
-    "Appenzeller",
-    "EntleBucher",
-    "boxer",
-    "bull mastiff",
-    "Tibetan mastiff",
-    "French bulldog",
-    "Great Dane",
-    "Saint Bernard, St Bernard",
-    "Eskimo dog, husky",
-    "malamute, malemute, Alaskan malamute",
-    "Siberian husky",
-    "dalmatian, coach dog, carriage dog",
-    "affenpinscher, monkey pinscher, monkey dog",
-    "basenji",
-    "pug, pug-dog",
-    "Leonberg",
-    "Newfoundland, Newfoundland dog",
-    "Great Pyrenees",
-    "Samoyed, Samoyede",
-    "Pomeranian",
-    "chow, chow chow",
-    "keeshond",
-    "Brabancon griffon",
-    "Pembroke, Pembroke Welsh corgi",
-    "Cardigan, Cardigan Welsh corgi",
-    "toy poodle",
-    "miniature poodle",
-    "standard poodle",
-    "Mexican hairless",
-    "timber wolf, grey wolf, gray wolf, Canis lupus",
-    "white wolf, Arctic wolf, Canis lupus tundrarum",
-    "red wolf, maned wolf, Canis rufus, Canis niger",
-    "coyote, prairie wolf, brush wolf, Canis latrans",
-    "dingo, warrigal, warragal, Canis dingo",
-    "dhole, Cuon alpinus",
-    "African hunting dog, hyena dog, Cape hunting dog, Lycaon pictus",
-    "hyena, hyaena",
-    "red fox, Vulpes vulpes",
-    "kit fox, Vulpes macrotis",
-    "Arctic fox, white fox, Alopex lagopus",
-    "grey fox, gray fox, Urocyon cinereoargenteus",
-    "tabby, tabby cat",
-    "tiger cat",
-    "Persian cat",
-    "Siamese cat, Siamese",
-    "Egyptian cat",
-    "cougar, puma, catamount, mountain lion, painter, panther, Felis concolor",
-    "lynx, catamount",
-    "leopard, Panthera pardus",
-    "snow leopard, ounce, Panthera uncia",
-    "jaguar, panther, Panthera onca, Felis onca",
-    "lion, king of beasts, Panthera leo",
-    "tiger, Panthera tigris",
-    "cheetah, chetah, Acinonyx jubatus",
-    "brown bear, bruin, Ursus arctos",
-    "American black bear, black bear, Ursus americanus, Euarctos americanus",
-    "ice bear, polar bear, Ursus Maritimus, Thalarctos maritimus",
-    "sloth bear, Melursus ursinus, Ursus ursinus",
-    "mongoose",
-    "meerkat, mierkat",
-    "tiger beetle",
-    "ladybug, ladybeetle, lady beetle, ladybird, ladybird beetle",
-    "ground beetle, carabid beetle",
-    "long-horned beetle, longicorn, longicorn beetle",
-    "leaf beetle, chrysomelid",
-    "dung beetle",
-    "rhinoceros beetle",
-    "weevil",
-    "fly",
-    "bee",
-    "ant, emmet, pismire",
-    "grasshopper, hopper",
-    "cricket",
-    "walking stick, walkingstick, stick insect",
-    "cockroach, roach",
-    "mantis, mantid",
-    "cicada, cicala",
-    "leafhopper",
-    "lacewing, lacewing fly",
-    "dragonfly, darning needle, devil's darning needle, sewing needle, snake feeder, snake doctor, mosquito hawk, skeeter hawk",
-    "damselfly",
-    "admiral",
-    "ringlet, ringlet butterfly",
-    "monarch, monarch butterfly, milkweed butterfly, Danaus plexippus",
-    "cabbage butterfly",
-    "sulphur butterfly, sulfur butterfly",
-    "lycaenid, lycaenid butterfly",
-    "starfish, sea star",
-    "sea urchin",
-    "sea cucumber, holothurian",
-    "wood rabbit, cottontail, cottontail rabbit",
-    "hare",
-    "Angora, Angora rabbit",
-    "hamster",
-    "porcupine, hedgehog",
-    "fox squirrel, eastern fox squirrel, Sciurus niger",
-    "marmot",
-    "beaver",
-    "guinea pig, Cavia cobaya",
-    "sorrel",
-    "zebra",
-    "hog, pig, grunter, squealer, Sus scrofa",
-    "wild boar, boar, Sus scrofa",
-    "warthog",
-    "hippopotamus, hippo, river horse, Hippopotamus amphibius",
-    "ox",
-    "water buffalo, water ox, Asiatic buffalo, Bubalus bubalis",
-    "bison",
-    "ram, tup",
-    "bighorn, bighorn sheep, cimarron, Rocky Mountain bighorn, Rocky Mountain sheep, Ovis canadensis",
-    "ibex, Capra ibex",
-    "hartebeest",
-    "impala, Aepyceros melampus",
-    "gazelle",
-    "Arabian camel, dromedary, Camelus dromedarius",
-    "llama",
-    "weasel",
-    "mink",
-    "polecat, fitch, foulmart, foumart, Mustela putorius",
-    "black-footed ferret, ferret, Mustela nigripes",
-    "otter",
-    "skunk, polecat, wood pussy",
-    "badger",
-    "armadillo",
-    "three-toed sloth, ai, Bradypus tridactylus",
-    "orangutan, orang, orangutang, Pongo pygmaeus",
-    "gorilla, Gorilla gorilla",
-    "chimpanzee, chimp, Pan troglodytes",
-    "gibbon, Hylobates lar",
-    "siamang, Hylobates syndactylus, Symphalangus syndactylus",
-    "guenon, guenon monkey",
-    "patas, hussar monkey, Erythrocebus patas",
-    "baboon",
-    "macaque",
-    "langur",
-    "colobus, colobus monkey",
-    "proboscis monkey, Nasalis larvatus",
-    "marmoset",
-    "capuchin, ringtail, Cebus capucinus",
-    "howler monkey, howler",
-    "titi, titi monkey",
-    "spider monkey, Ateles geoffroyi",
-    "squirrel monkey, Saimiri sciureus",
-    "Madagascar cat, ring-tailed lemur, Lemur catta",
-    "indri, indris, Indri indri, Indri brevicaudatus",
-    "Indian elephant, Elephas maximus",
-    "African elephant, Loxodonta africana",
-    "lesser panda, red panda, panda, bear cat, cat bear, Ailurus fulgens",
-    "giant panda, panda, panda bear, coon bear, Ailuropoda melanoleuca",
-    "barracouta, snoek",
-    "eel",
-    "coho, cohoe, coho salmon, blue jack, silver salmon, Oncorhynchus kisutch",
-    "rock beauty, Holocanthus tricolor",
-    "anemone fish",
-    "sturgeon",
-    "gar, garfish, garpike, billfish, Lepisosteus osseus",
-    "lionfish",
-    "puffer, pufferfish, blowfish, globefish",
-    "abacus",
-    "abaya",
-    "academic gown, academic robe, judge's robe",
-    "accordion, piano accordion, squeeze box",
-    "acoustic guitar",
-    "aircraft carrier, carrier, flattop, attack aircraft carrier",
-    "airliner",
-    "airship, dirigible",
-    "altar",
-    "ambulance",
-    "amphibian, amphibious vehicle",
-    "analog clock",
-    "apiary, bee house",
-    "apron",
-    "ashcan, trash can, garbage can, wastebin, ash bin, ash-bin, ashbin, dustbin, trash barrel, trash bin",
-    "assault rifle, assault gun",
-    "backpack, back pack, knapsack, packsack, rucksack, haversack",
-    "bakery, bakeshop, bakehouse",
-    "balance beam, beam",
-    "balloon",
-    "ballpoint, ballpoint pen, ballpen, Biro",
-    "Band Aid",
-    "banjo",
-    "bannister, banister, balustrade, balusters, handrail",
-    "barbell",
-    "barber chair",
-    "barbershop",
-    "barn",
-    "barometer",
-    "barrel, cask",
-    "barrow, garden cart, lawn cart, wheelbarrow",
-    "baseball",
-    "basketball",
-    "bassinet",
-    "bassoon",
-    "bathing cap, swimming cap",
-    "bath towel",
-    "bathtub, bathing tub, bath, tub",
-    "beach wagon, station wagon, wagon, estate car, beach waggon, station waggon, waggon",
-    "beacon, lighthouse, beacon light, pharos",
-    "beaker",
-    "bearskin, busby, shako",
-    "beer bottle",
-    "beer glass",
-    "bell cote, bell cot",
-    "bib",
-    "bicycle-built-for-two, tandem bicycle, tandem",
-    "bikini, two-piece",
-    "binder, ring-binder",
-    "binoculars, field glasses, opera glasses",
-    "birdhouse",
-    "boathouse",
-    "bobsled, bobsleigh, bob",
-    "bolo tie, bolo, bola tie, bola",
-    "bonnet, poke bonnet",
-    "bookcase",
-    "bookshop, bookstore, bookstall",
-    "bottlecap",
-    "bow",
-    "bow tie, bow-tie, bowtie",
-    "brass, memorial tablet, plaque",
-    "brassiere, bra, bandeau",
-    "breakwater, groin, groyne, mole, bulwark, seawall, jetty",
-    "breastplate, aegis, egis",
-    "broom",
-    "bucket, pail",
-    "buckle",
-    "bulletproof vest",
-    "bullet train, bullet",
-    "butcher shop, meat market",
-    "cab, hack, taxi, taxicab",
-    "caldron, cauldron",
-    "candle, taper, wax light",
-    "cannon",
-    "canoe",
-    "can opener, tin opener",
-    "cardigan",
-    "car mirror",
-    "carousel, carrousel, merry-go-round, roundabout, whirligig",
-    "carpenter's kit, tool kit",
-    "carton",
-    "car wheel",
-    "cash machine, cash dispenser, automated teller machine, automatic teller machine, automated teller, automatic teller, ATM",
-    "cassette",
-    "cassette player",
-    "castle",
-    "catamaran",
-    "CD player",
-    "cello, violoncello",
-    "cellular telephone, cellular phone, cellphone, cell, mobile phone",
-    "chain",
-    "chainlink fence",
-    "chain mail, ring mail, mail, chain armor, chain armour, ring armor, ring armour",
-    "chain saw, chainsaw",
-    "chest",
-    "chiffonier, commode",
-    "chime, bell, gong",
-    "china cabinet, china closet",
-    "Christmas stocking",
-    "church, church building",
-    "cinema, movie theater, movie theatre, movie house, picture palace",
-    "cleaver, meat cleaver, chopper",
-    "cliff dwelling",
-    "cloak",
-    "clog, geta, patten, sabot",
-    "cocktail shaker",
-    "coffee mug",
-    "coffeepot",
-    "coil, spiral, volute, whorl, helix",
-    "combination lock",
-    "computer keyboard, keypad",
-    "confectionery, confectionary, candy store",
-    "container ship, containership, container vessel",
-    "convertible",
-    "corkscrew, bottle screw",
-    "cornet, horn, trumpet, trump",
-    "cowboy boot",
-    "cowboy hat, ten-gallon hat",
-    "cradle",
-    "crane",
-    "crash helmet",
-    "crate",
-    "crib, cot",
-    "Crock Pot",
-    "croquet ball",
-    "crutch",
-    "cuirass",
-    "dam, dike, dyke",
-    "desk",
-    "desktop computer",
-    "dial telephone, dial phone",
-    "diaper, nappy, napkin",
-    "digital clock",
-    "digital watch",
-    "dining table, board",
-    "dishrag, dishcloth",
-    "dishwasher, dish washer, dishwashing machine",
-    "disk brake, disc brake",
-    "dock, dockage, docking facility",
-    "dogsled, dog sled, dog sleigh",
-    "dome",
-    "doormat, welcome mat",
-    "drilling platform, offshore rig",
-    "drum, membranophone, tympan",
-    "drumstick",
-    "dumbbell",
-    "Dutch oven",
-    "electric fan, blower",
-    "electric guitar",
-    "electric locomotive",
-    "entertainment center",
-    "envelope",
-    "espresso maker",
-    "face powder",
-    "feather boa, boa",
-    "file, file cabinet, filing cabinet",
-    "fireboat",
-    "fire engine, fire truck",
-    "fire screen, fireguard",
-    "flagpole, flagstaff",
-    "flute, transverse flute",
-    "folding chair",
-    "football helmet",
-    "forklift",
-    "fountain",
-    "fountain pen",
-    "four-poster",
-    "freight car",
-    "French horn, horn",
-    "frying pan, frypan, skillet",
-    "fur coat",
-    "garbage truck, dustcart",
-    "gasmask, respirator, gas helmet",
-    "gas pump, gasoline pump, petrol pump, island dispenser",
-    "goblet",
-    "go-kart",
-    "golf ball",
-    "golfcart, golf cart",
-    "gondola",
-    "gong, tam-tam",
-    "gown",
-    "grand piano, grand",
-    "greenhouse, nursery, glasshouse",
-    "grille, radiator grille",
-    "grocery store, grocery, food market, market",
-    "guillotine",
-    "hair slide",
-    "hair spray",
-    "half track",
-    "hammer",
-    "hamper",
-    "hand blower, blow dryer, blow drier, hair dryer, hair drier",
-    "hand-held computer, hand-held microcomputer",
-    "handkerchief, hankie, hanky, hankey",
-    "hard disc, hard disk, fixed disk",
-    "harmonica, mouth organ, harp, mouth harp",
-    "harp",
-    "harvester, reaper",
-    "hatchet",
-    "holster",
-    "home theater, home theatre",
-    "honeycomb",
-    "hook, claw",
-    "hoopskirt, crinoline",
-    "horizontal bar, high bar",
-    "horse cart, horse-cart",
-    "hourglass",
-    "iPod",
-    "iron, smoothing iron",
-    "jack-o'-lantern",
-    "jean, blue jean, denim",
-    "jeep, landrover",
-    "jersey, T-shirt, tee shirt",
-    "jigsaw puzzle",
-    "jinrikisha, ricksha, rickshaw",
-    "joystick",
-    "kimono",
-    "knee pad",
-    "knot",
-    "lab coat, laboratory coat",
-    "ladle",
-    "lampshade, lamp shade",
-    "laptop, laptop computer",
-    "lawn mower, mower",
-    "lens cap, lens cover",
-    "letter opener, paper knife, paperknife",
-    "library",
-    "lifeboat",
-    "lighter, light, igniter, ignitor",
-    "limousine, limo",
-    "liner, ocean liner",
-    "lipstick, lip rouge",
-    "Loafer",
-    "lotion",
-    "loudspeaker, speaker, speaker unit, loudspeaker system, speaker system",
-    "loupe, jeweler's loupe",
-    "lumbermill, sawmill",
-    "magnetic compass",
-    "mailbag, postbag",
-    "mailbox, letter box",
-    "maillot",
-    "maillot, tank suit",
-    "manhole cover",
-    "maraca",
-    "marimba, xylophone",
-    "mask",
-    "matchstick",
-    "maypole",
-    "maze, labyrinth",
-    "measuring cup",
-    "medicine chest, medicine cabinet",
-    "megalith, megalithic structure",
-    "microphone, mike",
-    "microwave, microwave oven",
-    "military uniform",
-    "milk can",
-    "minibus",
-    "miniskirt, mini",
-    "minivan",
-    "missile",
-    "mitten",
-    "mixing bowl",
-    "mobile home, manufactured home",
-    "Model T",
-    "modem",
-    "monastery",
-    "monitor",
-    "moped",
-    "mortar",
-    "mortarboard",
-    "mosque",
-    "mosquito net",
-    "motor scooter, scooter",
-    "mountain bike, all-terrain bike, off-roader",
-    "mountain tent",
-    "mouse, computer mouse",
-    "mousetrap",
-    "moving van",
-    "muzzle",
-    "nail",
-    "neck brace",
-    "necklace",
-    "nipple",
-    "notebook, notebook computer",
-    "obelisk",
-    "oboe, hautboy, hautbois",
-    "ocarina, sweet potato",
-    "odometer, hodometer, mileometer, milometer",
-    "oil filter",
-    "organ, pipe organ",
-    "oscilloscope, scope, cathode-ray oscilloscope, CRO",
-    "overskirt",
-    "oxcart",
-    "oxygen mask",
-    "packet",
-    "paddle, boat paddle",
-    "paddlewheel, paddle wheel",
-    "padlock",
-    "paintbrush",
-    "pajama, pyjama, pj's, jammies",
-    "palace",
-    "panpipe, pandean pipe, syrinx",
-    "paper towel",
-    "parachute, chute",
-    "parallel bars, bars",
-    "park bench",
-    "parking meter",
-    "passenger car, coach, carriage",
-    "patio, terrace",
-    "pay-phone, pay-station",
-    "pedestal, plinth, footstall",
-    "pencil box, pencil case",
-    "pencil sharpener",
-    "perfume, essence",
-    "Petri dish",
-    "photocopier",
-    "pick, plectrum, plectron",
-    "pickelhaube",
-    "picket fence, paling",
-    "pickup, pickup truck",
-    "pier",
-    "piggy bank, penny bank",
-    "pill bottle",
-    "pillow",
-    "ping-pong ball",
-    "pinwheel",
-    "pirate, pirate ship",
-    "pitcher, ewer",
-    "plane, carpenter's plane, woodworking plane",
-    "planetarium",
-    "plastic bag",
-    "plate rack",
-    "plow, plough",
-    "plunger, plumber's helper",
-    "Polaroid camera, Polaroid Land camera",
-    "pole",
-    "police van, police wagon, paddy wagon, patrol wagon, wagon, black Maria",
-    "poncho",
-    "pool table, billiard table, snooker table",
-    "pop bottle, soda bottle",
-    "pot, flowerpot",
-    "potter's wheel",
-    "power drill",
-    "prayer rug, prayer mat",
-    "printer",
-    "prison, prison house",
-    "projectile, missile",
-    "projector",
-    "puck, hockey puck",
-    "punching bag, punch bag, punching ball, punchball",
-    "purse",
-    "quill, quill pen",
-    "quilt, comforter, comfort, puff",
-    "racer, race car, racing car",
-    "racket, racquet",
-    "radiator",
-    "radio, wireless",
-    "radio telescope, radio reflector",
-    "rain barrel",
-    "recreational vehicle, RV, R.V.",
-    "reel",
-    "reflex camera",
-    "refrigerator, icebox",
-    "remote control, remote",
-    "restaurant, eating house, eating place, eatery",
-    "revolver, six-gun, six-shooter",
-    "rifle",
-    "rocking chair, rocker",
-    "rotisserie",
-    "rubber eraser, rubber, pencil eraser",
-    "rugby ball",
-    "rule, ruler",
-    "running shoe",
-    "safe",
-    "safety pin",
-    "saltshaker, salt shaker",
-    "sandal",
-    "sarong",
-    "sax, saxophone",
-    "scabbard",
-    "scale, weighing machine",
-    "school bus",
-    "schooner",
-    "scoreboard",
-    "screen, CRT screen",
-    "screw",
-    "screwdriver",
-    "seat belt, seatbelt",
-    "sewing machine",
-    "shield, buckler",
-    "shoe shop, shoe-shop, shoe store",
-    "shoji",
-    "shopping basket",
-    "shopping cart",
-    "shovel",
-    "shower cap",
-    "shower curtain",
-    "ski",
-    "ski mask",
-    "sleeping bag",
-    "slide rule, slipstick",
-    "sliding door",
-    "slot, one-armed bandit",
-    "snorkel",
-    "snowmobile",
-    "snowplow, snowplough",
-    "soap dispenser",
-    "soccer ball",
-    "sock",
-    "solar dish, solar collector, solar furnace",
-    "sombrero",
-    "soup bowl",
-    "space bar",
-    "space heater",
-    "space shuttle",
-    "spatula",
-    "speedboat",
-    "spider web, spider's web",
-    "spindle",
-    "sports car, sport car",
-    "spotlight, spot",
-    "stage",
-    "steam locomotive",
-    "steel arch bridge",
-    "steel drum",
-    "stethoscope",
-    "stole",
-    "stone wall",
-    "stopwatch, stop watch",
-    "stove",
-    "strainer",
-    "streetcar, tram, tramcar, trolley, trolley car",
-    "stretcher",
-    "studio couch, day bed",
-    "stupa, tope",
-    "submarine, pigboat, sub, U-boat",
-    "suit, suit of clothes",
-    "sundial",
-    "sunglass",
-    "sunglasses, dark glasses, shades",
-    "sunscreen, sunblock, sun blocker",
-    "suspension bridge",
-    "swab, swob, mop",
-    "sweatshirt",
-    "swimming trunks, bathing trunks",
-    "swing",
-    "switch, electric switch, electrical switch",
-    "syringe",
-    "table lamp",
-    "tank, army tank, armored combat vehicle, armoured combat vehicle",
-    "tape player",
-    "teapot",
-    "teddy, teddy bear",
-    "television, television system",
-    "tennis ball",
-    "thatch, thatched roof",
-    "theater curtain, theatre curtain",
-    "thimble",
-    "thresher, thrasher, threshing machine",
-    "throne",
-    "tile roof",
-    "toaster",
-    "tobacco shop, tobacconist shop, tobacconist",
-    "toilet seat",
-    "torch",
-    "totem pole",
-    "tow truck, tow car, wrecker",
-    "toyshop",
-    "tractor",
-    "trailer truck, tractor trailer, trucking rig, rig, articulated lorry, semi",
-    "tray",
-    "trench coat",
-    "tricycle, trike, velocipede",
-    "trimaran",
-    "tripod",
-    "triumphal arch",
-    "trolleybus, trolley coach, trackless trolley",
-    "trombone",
-    "tub, vat",
-    "turnstile",
-    "typewriter keyboard",
-    "umbrella",
-    "unicycle, monocycle",
-    "upright, upright piano",
-    "vacuum, vacuum cleaner",
-    "vase",
-    "vault",
-    "velvet",
-    "vending machine",
-    "vestment",
-    "viaduct",
-    "violin, fiddle",
-    "volleyball",
-    "waffle iron",
-    "wall clock",
-    "wallet, billfold, notecase, pocketbook",
-    "wardrobe, closet, press",
-    "warplane, military plane",
-    "washbasin, handbasin, washbowl, lavabo, wash-hand basin",
-    "washer, automatic washer, washing machine",
-    "water bottle",
-    "water jug",
-    "water tower",
-    "whiskey jug",
-    "whistle",
-    "wig",
-    "window screen",
-    "window shade",
-    "Windsor tie",
-    "wine bottle",
-    "wing",
-    "wok",
-    "wooden spoon",
-    "wool, woolen, woollen",
-    "worm fence, snake fence, snake-rail fence, Virginia fence",
-    "wreck",
-    "yawl",
-    "yurt",
-    "web site, website, internet site, site",
-    "comic book",
-    "crossword puzzle, crossword",
-    "street sign",
-    "traffic light, traffic signal, stoplight",
-    "book jacket, dust cover, dust jacket, dust wrapper",
-    "menu",
-    "plate",
-    "guacamole",
-    "consomme",
-    "hot pot, hotpot",
-    "trifle",
-    "ice cream, icecream",
-    "ice lolly, lolly, lollipop, popsicle",
-    "French loaf",
-    "bagel, beigel",
-    "pretzel",
-    "cheeseburger",
-    "hotdog, hot dog, red hot",
-    "mashed potato",
-    "head cabbage",
-    "broccoli",
-    "cauliflower",
-    "zucchini, courgette",
-    "spaghetti squash",
-    "acorn squash",
-    "butternut squash",
-    "cucumber, cuke",
-    "artichoke, globe artichoke",
-    "bell pepper",
-    "cardoon",
-    "mushroom",
-    "Granny Smith",
-    "strawberry",
-    "orange",
-    "lemon",
-    "fig",
-    "pineapple, ananas",
-    "banana",
-    "jackfruit, jak, jack",
-    "custard apple",
-    "pomegranate",
-    "hay",
-    "carbonara",
-    "chocolate sauce, chocolate syrup",
-    "dough",
-    "meat loaf, meatloaf",
-    "pizza, pizza pie",
-    "potpie",
-    "burrito",
-    "red wine",
-    "espresso",
-    "cup",
-    "eggnog",
-    "alp",
-    "bubble",
-    "cliff, drop, drop-off",
-    "coral reef",
-    "geyser",
-    "lakeside, lakeshore",
-    "promontory, headland, head, foreland",
-    "sandbar, sand bar",
-    "seashore, coast, seacoast, sea-coast",
-    "valley, vale",
-    "volcano",
-    "ballplayer, baseball player",
-    "groom, bridegroom",
-    "scuba diver",
-    "rapeseed",
-    "daisy",
-    "yellow lady's slipper, yellow lady-slipper, Cypripedium calceolus, Cypripedium parviflorum",
-    "corn",
-    "acorn",
-    "hip, rose hip, rosehip",
-    "buckeye, horse chestnut, conker",
-    "coral fungus",
-    "agaric",
-    "gyromitra",
-    "stinkhorn, carrion fungus",
-    "earthstar",
-    "hen-of-the-woods, hen of the woods, Polyporus frondosus, Grifola frondosa",
-    "bolete",
-    "ear, spike, capitulum",
-    "toilet tissue, toilet paper, bathroom tissue",
-];
diff --git a/src/settings.rs b/src/settings.rs
index d04eedd..3e7fe05 100644
--- a/src/settings.rs
+++ b/src/settings.rs
@@ -24,6 +24,9 @@ pub struct Settings {
     /// Path for ViT image model
     pub vit_model_path: Option<PathBuf>,
 
+    /// Path for ViT image model config.json
+    pub vit_model_config_path: Option<PathBuf>,
+
     /// Webhook api endpoint
     pub webhook_url: Option<String>,