georgedavila/bart-large-mnli-classifier 📝 → 📝

▶️ 4.3K runs 📅 Jan 2024 ⚙️ Cog 0.8.6 🔗 GitHub

text-classification zero-shot-classification

About

Zero-shot classifier which classifies text into categories of your choosing. Returns a dictionary of the most likely class and all class likelihoods.

Example Output

Output

{"allClasses":{"Cooking Instructions":0.9597448110580444,"Question about Astronomy":0.04025513678789139},"mostLikelyClass":"Cooking Instructions"}

Performance Metrics

12.64s Prediction Time

125.68s Total Time

All Input Parameters

{
  "labels": "Cooking Instructions, Question about Astronomy",
  "text2classify": "Add salt to boiling water to prevent pasta from sticking together"
}

Input Parameters

labels Type: stringDefault: Cooking Instructions, Question about Astronomy: Possible class names (comma-separated). This is a zero-shot classifier so you can try any label you'd like. The model will output the top label under key 'mostLikelyClass'.
text2classify Type: stringDefault: Add salt to boiling water to prevent pasta from sticking together: Text you want to classify.

Output Schema

Output

Type: string

Example Execution Logs

config.json:   0%|          | 0.00/1.15k [00:00<?, ?B/s]
config.json: 100%|██████████| 1.15k/1.15k [00:00<00:00, 5.85MB/s]
model.safetensors:   0%|          | 0.00/1.63G [00:00<?, ?B/s]
model.safetensors:   1%|          | 10.5M/1.63G [00:00<00:39, 41.3MB/s]
model.safetensors:   2%|▏         | 31.5M/1.63G [00:00<00:15, 100MB/s] 
model.safetensors:   3%|▎         | 52.4M/1.63G [00:00<00:12, 129MB/s]
model.safetensors:   5%|▍         | 73.4M/1.63G [00:00<00:11, 131MB/s]
model.safetensors:   6%|▌         | 94.4M/1.63G [00:00<00:10, 140MB/s]
model.safetensors:   7%|▋         | 115M/1.63G [00:00<00:09, 158MB/s] 
model.safetensors:   8%|▊         | 136M/1.63G [00:00<00:08, 169MB/s]
model.safetensors:  10%|▉         | 157M/1.63G [00:01<00:08, 167MB/s]
model.safetensors:  11%|█         | 178M/1.63G [00:01<00:08, 178MB/s]
model.safetensors:  12%|█▏        | 199M/1.63G [00:01<00:08, 177MB/s]
model.safetensors:  14%|█▎        | 220M/1.63G [00:01<00:07, 183MB/s]
model.safetensors:  15%|█▍        | 241M/1.63G [00:01<00:07, 186MB/s]
model.safetensors:  16%|█▌        | 262M/1.63G [00:01<00:07, 189MB/s]
model.safetensors:  17%|█▋        | 283M/1.63G [00:01<00:06, 193MB/s]
model.safetensors:  19%|█▉        | 315M/1.63G [00:01<00:06, 199MB/s]
model.safetensors:  21%|██        | 336M/1.63G [00:02<00:06, 198MB/s]
model.safetensors:  22%|██▏       | 357M/1.63G [00:02<00:06, 200MB/s]
model.safetensors:  23%|██▎       | 377M/1.63G [00:02<00:06, 197MB/s]
model.safetensors:  24%|██▍       | 398M/1.63G [00:02<00:06, 197MB/s]
model.safetensors:  26%|██▋       | 430M/1.63G [00:02<00:05, 202MB/s]
model.safetensors:  28%|██▊       | 461M/1.63G [00:02<00:05, 206MB/s]
model.safetensors:  30%|███       | 493M/1.63G [00:02<00:05, 209MB/s]
model.safetensors:  32%|███▏      | 524M/1.63G [00:02<00:05, 208MB/s]
model.safetensors:  33%|███▎      | 545M/1.63G [00:03<00:05, 206MB/s]
model.safetensors:  35%|███▍      | 566M/1.63G [00:03<00:05, 203MB/s]
model.safetensors:  36%|███▌      | 587M/1.63G [00:03<00:05, 202MB/s]
model.safetensors:  37%|███▋      | 608M/1.63G [00:03<00:05, 202MB/s]
model.safetensors:  39%|███▊      | 629M/1.63G [00:03<00:04, 203MB/s]
model.safetensors:  41%|████      | 661M/1.63G [00:03<00:04, 206MB/s]
model.safetensors:  42%|████▏     | 682M/1.63G [00:03<00:04, 205MB/s]
model.safetensors:  43%|████▎     | 703M/1.63G [00:03<00:04, 203MB/s]
model.safetensors:  44%|████▍     | 724M/1.63G [00:03<00:05, 178MB/s]
model.safetensors:  46%|████▌     | 744M/1.63G [00:04<00:04, 186MB/s]
model.safetensors:  48%|████▊     | 776M/1.63G [00:04<00:04, 193MB/s]
model.safetensors:  49%|████▉     | 797M/1.63G [00:04<00:04, 195MB/s]
model.safetensors:  50%|█████     | 818M/1.63G [00:04<00:04, 198MB/s]
model.safetensors:  52%|█████▏    | 849M/1.63G [00:04<00:03, 203MB/s]
model.safetensors:  54%|█████▍    | 881M/1.63G [00:04<00:03, 201MB/s]
model.safetensors:  55%|█████▌    | 902M/1.63G [00:04<00:03, 200MB/s]
model.safetensors:  57%|█████▋    | 923M/1.63G [00:04<00:03, 199MB/s]
model.safetensors:  58%|█████▊    | 944M/1.63G [00:05<00:03, 202MB/s]
model.safetensors:  60%|█████▉    | 975M/1.63G [00:05<00:03, 204MB/s]
model.safetensors:  61%|██████    | 996M/1.63G [00:05<00:03, 204MB/s]
model.safetensors:  62%|██████▏   | 1.02G/1.63G [00:05<00:03, 200MB/s]
model.safetensors:  64%|██████▎   | 1.04G/1.63G [00:05<00:02, 200MB/s]
model.safetensors:  66%|██████▌   | 1.07G/1.63G [00:05<00:02, 205MB/s]
model.safetensors:  67%|██████▋   | 1.09G/1.63G [00:05<00:02, 191MB/s]
model.safetensors:  69%|██████▉   | 1.12G/1.63G [00:05<00:02, 199MB/s]
model.safetensors:  70%|███████   | 1.14G/1.63G [00:06<00:02, 200MB/s]
model.safetensors:  71%|███████▏  | 1.16G/1.63G [00:06<00:02, 201MB/s]
model.safetensors:  73%|███████▎  | 1.18G/1.63G [00:06<00:02, 204MB/s]
model.safetensors:  75%|███████▍  | 1.22G/1.63G [00:06<00:01, 207MB/s]
model.safetensors:  77%|███████▋  | 1.25G/1.63G [00:06<00:01, 209MB/s]
model.safetensors:  78%|███████▊  | 1.27G/1.63G [00:06<00:01, 182MB/s]
model.safetensors:  79%|███████▉  | 1.29G/1.63G [00:06<00:01, 185MB/s]
model.safetensors:  80%|████████  | 1.31G/1.63G [00:06<00:01, 172MB/s]
model.safetensors:  82%|████████▏ | 1.33G/1.63G [00:07<00:01, 176MB/s]
model.safetensors:  83%|████████▎ | 1.35G/1.63G [00:07<00:01, 183MB/s]
model.safetensors:  84%|████████▍ | 1.37G/1.63G [00:07<00:01, 189MB/s]
model.safetensors:  86%|████████▌ | 1.41G/1.63G [00:07<00:01, 198MB/s]
model.safetensors:  88%|████████▊ | 1.44G/1.63G [00:07<00:01, 192MB/s]
model.safetensors:  90%|█████████ | 1.47G/1.63G [00:07<00:00, 197MB/s]
model.safetensors:  91%|█████████▏| 1.49G/1.63G [00:07<00:00, 199MB/s]
model.safetensors:  93%|█████████▎| 1.51G/1.63G [00:07<00:00, 189MB/s]
model.safetensors:  94%|█████████▍| 1.53G/1.63G [00:08<00:00, 189MB/s]
model.safetensors:  95%|█████████▌| 1.55G/1.63G [00:08<00:00, 193MB/s]
model.safetensors:  97%|█████████▋| 1.57G/1.63G [00:08<00:00, 178MB/s]
model.safetensors:  98%|█████████▊| 1.59G/1.63G [00:08<00:00, 185MB/s]
model.safetensors:  99%|█████████▉| 1.61G/1.63G [00:08<00:00, 170MB/s]
model.safetensors: 100%|██████████| 1.63G/1.63G [00:08<00:00, 186MB/s]
tokenizer_config.json:   0%|          | 0.00/26.0 [00:00<?, ?B/s]
tokenizer_config.json: 100%|██████████| 26.0/26.0 [00:00<00:00, 78.8kB/s]
vocab.json:   0%|          | 0.00/899k [00:00<?, ?B/s]
vocab.json: 100%|██████████| 899k/899k [00:00<00:00, 9.29MB/s]
merges.txt:   0%|          | 0.00/456k [00:00<?, ?B/s]
merges.txt: 100%|██████████| 456k/456k [00:00<00:00, 61.4MB/s]
tokenizer.json:   0%|          | 0.00/1.36M [00:00<?, ?B/s]
tokenizer.json: 100%|██████████| 1.36M/1.36M [00:00<00:00, 8.29MB/s]
tokenizer.json: 100%|██████████| 1.36M/1.36M [00:00<00:00, 8.17MB/s]

Version Details

Version ID: d929487cf059f96a17752ebe55ae5a85b2e8be6cd627078e49c6caa2fd4213db
Version Created: January 3, 2024

Run on Replicate →