ferdmartin commited on
Commit
d411fb6
·
1 Parent(s): 08c3e89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -96,8 +96,8 @@ def main():
96
 
97
  # Streamlit app:
98
 
99
- models_available = {"Logistic Regression":"models/baseline_model_lr2.joblib",
100
- "Naive Bayes": "models/baseline_model_nb2.joblib",
101
  "DistilBERT-based model (BERT light)": "ferdmartin/HF_DistilBertBasedModelAppDocs",
102
  "BERT-based model": "ferdmartin/HF_BertBasedModelAppDocs"
103
  }
 
96
 
97
  # Streamlit app:
98
 
99
+ models_available = {"Logistic Regression":"models/baseline_model_lr_longer.joblib",
100
+ "Naive Bayes": "models/baseline_model_nb_longer.joblib",
101
  "DistilBERT-based model (BERT light)": "ferdmartin/HF_DistilBertBasedModelAppDocs",
102
  "BERT-based model": "ferdmartin/HF_BertBasedModelAppDocs"
103
  }