Update app.py

app.py

@@ -105,7 +105,7 @@ def check_if_passed(model):
   if model["best_result"] >= model["min_result"]:
     model["passed"] = True
 
-def test_(hf_username):
+def certification(hf_username):
   results_certification = [
       {
           "unit": "Unit 1",
@@ -198,7 +198,7 @@ def test_(hf_username):
           "passed": False
     },
       {
-          "unit": "Unit 8 Part 1",
+          "unit": "Unit 8 PI",
           "env": "GodotRL-JumperHard",
           "library": "cleanrl",
           "min_result": -100,
@@ -207,7 +207,7 @@ def test_(hf_username):
           "passed": False
     },
       {
-          "unit": "Unit 8 Part 2",
+          "unit": "Unit 8 PII",
           "env": "Vizdoom-Battle",
           "library": "cleanrl",
           "min_result": -100,
@@ -219,7 +219,7 @@ def test_(hf_username):
   for unit in results_certification:
     # Get user model
     user_models = get_user_models(hf_username, unit['env'], unit['library'])
-    print(user_models)
+
     # Calculate the best result and get the best_model_id
     best_result, best_model_id = calculate_best_result(user_models)
 
@@ -233,8 +233,8 @@ def test_(hf_username):
     
   print(results_certification)
  
-  df = pd.DataFrame (results_certification)
-
+  df = pd.DataFrame(results_certification)
+  df = df[['passed', 'unit1', 'env', 'min_result', 'best_result', 'best_model_id']]
   return df
 
 
@@ -256,7 +256,7 @@ with gr.Blocks() as demo:
     hf_username = gr.Textbox(placeholder="ThomasSimonini", label="Your Hugging Face Username")
     #email = gr.Textbox(placeholder="thomas.simonini@huggingface.co", label="Your Email (to receive your certificate)")
     check_progress_button = gr.Button(value="Check my progress")
-    output = gr.components.Dataframe(value= test_(hf_username), headers=["Unit", "Environment", "Library", "Baseline", "Your best result", "Your best model id", "Pass?"], datatype=["markdown", "markdown", "markdown", "number", "number", "markdown", "bool"])
+    output = gr.components.Dataframe(value= certification(hf_username), headers=["Unit", "Environment", "Library", "Baseline", "Your best result", "Your best model id", "Pass?"], datatype=["markdown", "markdown", "markdown", "number", "number", "markdown", "bool"])
     check_progress_button.click(fn=test_, inputs=hf_username, outputs=output)
 
 demo.launch()