Update app.py

app.py

@@ -40,7 +40,7 @@ def main():
         
         return text.replace("REDACTED", "").lower().replace("  "," ").replace("[Name]", "").replace("[your name]", "").replace("\n your name", "").\
                                 replace("dear admissions committee,", "").replace("sincerely,","").\
-                                replace("[university's name]","fordham").replace("dear sir/madam,","").\
+                                replace("[university's name]").replace("dear sir/madam,","").\
                                 replace("– statement of intent  ","").\
                                 replace('program: master of science in data analytics  name of applicant:    ',"").\
                                 replace("data analytics", "data science").replace("| \u200b","").\
@@ -79,7 +79,7 @@ def main():
     
     @st.cache(allow_output_mutation=True, suppress_st_warning=True)
     def load_tokenizer(option):
-        if option == "BERT-based model":
+        if option in ("BERT Generated", "BERT Revised"):
             tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased", padding='max_length', max_length=512, truncation=True)
         else:
             tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased", padding='max_length', max_length=512, truncation=True)
@@ -87,10 +87,14 @@ 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("GradApplicationDocsApp/HF_BertBasedModelAppDocs3").to(device)
+        if option == "BERT Generated":
+            model = HF_BertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/BERTGradGen").to(device)
+        elif option == "BERT Revised":
+            model = HF_BertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/BERTGradRevised").to(device)
+        elif option == "D-BERT Generated":
+            model = HF_BertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/DistilBERTGradGen").to(device)
         else:
-            model = HF_DistilBertBasedModelAppDocs.from_pretrained("GradApplicationDocsApp/HF_DistilBertBasedModelAppDocs3").to(device)
+            model = HF_DistilBertBasedModelAppDocs.from_pretrained("GradApplicationDocuments/DistilBERTGradRevised").to(device)
         return model
     
 
@@ -98,8 +102,12 @@ def main():
 
     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"
+                        "D-BERT Generated": "GradApplicationDocuments/DistilBERTGradGen",
+                        "BERT Generated": "GradApplicationDocuments/BERTGradGen",
+                        "LR Revised":"models/baseline_model_lr_revised.joblib", 
+                        "NB Revised": "models/baseline_model_nb_revised.joblib",
+                        "D-BERT Revised": "GradApplicationDocuments/DistilBERTGradRevised",
+                        "BERT Revised": "GradApplicationDocuments/BERTGradRevised",
                         }
 
     st.set_page_config(page_title="AI/Human GradAppDocs", page_icon="🤖", layout="wide")