Spaces:
Sleeping
Sleeping
Commit
·
d411fb6
1
Parent(s):
08c3e89
Update app.py
Browse files
app.py
CHANGED
@@ -96,8 +96,8 @@ def main():
|
|
96 |
|
97 |
# Streamlit app:
|
98 |
|
99 |
-
models_available = {"Logistic Regression":"models/
|
100 |
-
"Naive Bayes": "models/
|
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 |
}
|