ferdmartin commited on
Commit
f295a5b
·
1 Parent(s): 3e5dcae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/HF_DistilBertBasedModelAppDocs",
102
- "BERT-based model": "ferdmartin/HF_BertBasedModelAppDocs"
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")