Spaces:
Sleeping
Sleeping
Commit
·
f295a5b
1
Parent(s):
3e5dcae
Update app.py
Browse files
app.py
CHANGED
@@ -98,8 +98,8 @@ def main():
|
|
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/
|
102 |
-
"BERT-based model": "ferdmartin/
|
103 |
}
|
104 |
|
105 |
st.set_page_config(page_title="AI/Human GradAppDocs", page_icon="🤖", layout="wide")
|
|
|
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_DistilBertBasedModelAppDocs2",
|
102 |
+
"BERT-based model": "ferdmartin/HF_BertBasedModelAppDocs2"
|
103 |
}
|
104 |
|
105 |
st.set_page_config(page_title="AI/Human GradAppDocs", page_icon="🤖", layout="wide")
|