Update app.py

app.py

@@ -88,9 +88,9 @@ def main():
     @st.cache(allow_output_mutation=True, suppress_st_warning=True)
     def load_model(option):
         if option == "BERT-based model":
-            model = HF_BertBasedModelAppDocs.from_pretrained("ferdmartin/HF_BertBasedModelAppDocs").to(device)
+            model = HF_BertBasedModelAppDocs.from_pretrained("ferdmartin/HF_BertBasedModelAppDocs2").to(device)
         else:
-            model = HF_DistilBertBasedModelAppDocs.from_pretrained("ferdmartin/HF_DistilBertBasedModelAppDocs").to(device)
+            model = HF_DistilBertBasedModelAppDocs.from_pretrained("ferdmartin/HF_DistilBertBasedModelAppDocs2").to(device)
         return model
     
 
@@ -98,8 +98,8 @@ def main():
 
     models_available = {"Logistic Regression":"models/baseline_model_lr2.joblib", 
                         "Naive Bayes": "models/baseline_model_nb2.joblib",
-                        "DistilBERT-based model (BERT light)": "ferdmartin/HF_DistilBertBasedModelAppDocs2",
-                        "BERT-based model": "ferdmartin/HF_BertBasedModelAppDocs2"
+                        "DistilBERT-based model (BERT light)": "ferdmartin/HF_DistilBertBasedModelAppDocs",
+                        "BERT-based model": "ferdmartin/HF_BertBasedModelAppDocs"
                         }
 
     st.set_page_config(page_title="AI/Human GradAppDocs", page_icon="🤖", layout="wide")