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 |
}
|