Update app.py

changed model name

app.py

@@ -34,21 +34,27 @@ def process_file(model_name,inc_slider,progress=Progress(track_tqdm=True)):
     label_location=parent_location+"overallTestData/test_label.txt"
     # "ASTRA-FT-HGR-RANDOM10", "ASTRA-FT-FIRST10-WSKILLS", "ASTRA-FT-FIRST10-WTIME", "ASTRA-FT-FIRST10-WSKILLS_WTIME"
     checkpoint = "ratio_proportion_change3_2223/sch_largest_100-coded/output/"
-    if(model_name=="ASTRA-FT-HGR-RANDOM10"):
+    if(model_name=="ASTRA-FT-HGR"):
         finetune_task="highGRschool10"
         checkpoint += "highGRschool10/bert_fine_tuned.model.ep42"
+        model_task="ASTRA-FT-HGR-RANDOM10"
     elif(model_name== "ASTRA-FT-LGR" ):
         finetune_task="lowGRschoolAll"
+        model_task="ASTRA-FT-LGR"
     elif(model_name=="ASTRA-FT-FULL"):
         finetune_task="fullTest"
-    elif(model_name in ["ASTRA-FT-FIRST10-WSKILLS", "ASTRA-FT-FIRST10-WTIME", "ASTRA-FT-FIRST10-WSKILLS_WTIME"]):
+        model_task="ASTRA-FT-FULL"
+    elif(model_name in ["ASTRA-FT-Skills", "ASTRA-FT-Temporal", "ASTRA-FT-Unified"]):
         finetune_task="first10"
-        if model_name == "ASTRA-FT-FIRST10-WSKILLS":
+        if model_name == "ASTRA-FT-Skills":
             checkpoint += "first10/bert_fine_tuned.model.first10%.wskills.ep24"
-        elif model_name == "ASTRA-FT-FIRST10-WTIME":
+            model_task="ASTRA-FT-FIRST10-WSKILLS"
+        elif model_name == "ASTRA-FT-Temporal":
             checkpoint += "first10/bert_fine_tuned.model.first10%.wfaopttime.wttime.wttopttime.wttnoopttime.ep23"
-        elif model_name == "ASTRA-FT-FIRST10-WSKILLS_WTIME":
+            model_task="ASTRA-FT-FIRST10-WTIME"
+        elif model_name == "ASTRA-FT-Unified":
             checkpoint += "first10/bert_fine_tuned.model.first10%.wskills.wfaopttime.wttime.wttopttime.wttnoopttime.ep40"
+            model_task="ASTRA-FT-FIRST10-WSKILLS_WTIME"
     else:
         finetune_task=None
     # Load the test_info file and the graduation rate file
@@ -132,7 +138,7 @@ def process_file(model_name,inc_slider,progress=Progress(track_tqdm=True)):
     subprocess.run([
         "python", "new_test_saved_finetuned_model.py",
         "-workspace_name", "ratio_proportion_change3_2223/sch_largest_100-coded",
-        "-model_name", model_name,
+        "-model_name", model_task,
         "-finetune_task", finetune_task,
         "-test_dataset_path","../../../../fileHandler/selected_rows.txt",
         # "-test_label_path","../../../../train_label.txt",
@@ -778,7 +784,7 @@ def process_file(model_name,inc_slider,progress=Progress(track_tqdm=True)):
 
 # models = ["ASTRA-FT-HGR", "ASTRA-FT-LGR", "ASTRA-FT-FULL"]
 # models = ["ASTRA-FT-HGR", "ASTRA-FT-FULL"]
-models = ["ASTRA-FT-HGR-RANDOM10", "ASTRA-FT-FIRST10-WSKILLS", "ASTRA-FT-FIRST10-WTIME", "ASTRA-FT-FIRST10-WSKILLS_WTIME"]
+models = ["ASTRA-FT-HGR", "ASTRA-FT-Skills", "ASTRA-FT-Temporal", "ASTRA-FT-Unified"]
 
 content = """
 <h1 style="color: black;">A S T R A</h1>