add epoch

app.py

@@ -29,12 +29,12 @@ def baseline_load_data(model,version,metrics):
     
     # we only want specific columns and in a specific order
     
-    column_names = ["Method", "Submitted By",
+    column_names = ["Method", "Submitted By", "Epoch",
                     "Model Utility", "Forget Quality",
                     "ROUGE Real Authors", "Truth Ratio Real Authors", "Prob. Real Authors", 
                     "ROUGE Real World", "Truth Ratio Real World", "Prob. Real World", 
                     "ROUGE Retain", "Truth Ratio Retain", "Prob. Retain", 
-                    "ROUGE Forget", "Truth Ratio Forget", "Prob. Forget", "Epoch"
+                    "ROUGE Forget", "Truth Ratio Forget", "Prob. Forget",
                     ]
     #based on the metrics, remove the columns that are not needed
     if "ROUGE" not in metrics: