Update app.py

app.py

@@ -96,10 +96,10 @@ def main():
 
     # Streamlit app:
 
-    models_available = {"Logistic Regression":"models/baseline_model_lr.joblib", 
-                        "Naive Bayes": "models/baseline_model_nb.joblib",
-                        "DistilBERT-based model (BERT light)": "GradApplicationDocsApp/HF_DistilBertBasedModelAppDocs3",
-                        "BERT-based model": "GradApplicationDocsApp/HF_BertBasedModelAppDocs3"
+    models_available = {"LR Generated":"models/baseline_model_lr_generated.joblib", 
+                        "NB Generated": "models/baseline_model_nb_generated.joblib",
+                        "D-BERT Generated": "GradApplicationDocuments/BERTGradGen",
+                        "BERT Generated": "GradApplicationDocuments/DistilBERTGradGen"
                         }
 
     st.set_page_config(page_title="AI/Human GradAppDocs", page_icon="🤖", layout="wide")