tbitai commited on
Commit
5ab9f7f
·
verified ·
1 Parent(s): ebc9c45

Fix order of model names and display names

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,9 +8,9 @@ import numpy as np
8
  # Load models
9
 
10
  MODELS = [
11
- (BAYES := "bayes-enron1-spam", "Bayes Enron1 spam"),
12
- (NN := "nn-enron1-spam", "NN Enron1 spam"),
13
- (LLM := "gisty-enron1-spam", "GISTy Enron1 spam"),
14
  ]
15
 
16
  model_probs_path = hf_hub_download(repo_id=f"tbitai/{BAYES}", filename="probs.json")
 
8
  # Load models
9
 
10
  MODELS = [
11
+ ("Bayes Enron1 spam", BAYES := "bayes-enron1-spam"),
12
+ ("NN Enron1 spam", NN := "nn-enron1-spam"),
13
+ ("GISTy Enron1 spam", LLM := "gisty-enron1-spam"),
14
  ]
15
 
16
  model_probs_path = hf_hub_download(repo_id=f"tbitai/{BAYES}", filename="probs.json")