Update app.py

app.py

@@ -206,16 +206,7 @@ def certification(hf_username, first_name, last_name):
     },
       {
           "unit": "Unit 6",
-          "env": "AntBulletEnv-v0",
-          "library": "stable-baselines3",
-          "min_result": 650,
-          "best_result": 0,
-          "best_model_id": "",
-          "passed_": False
-    },
-      {
-          "unit": "Unit 6",
-          "env": "PandaReachDense-v2",
+          "env": "PandaReachDense",
           "library": "stable-baselines3",
           "min_result": -3.5,
           "best_result": 0,
@@ -251,6 +242,12 @@ def certification(hf_username, first_name, last_name):
     },
   ] 
   for unit in results_certification:
+    if unit["unit"] != "Unit 6":
+        # Since Unit 6 can use PandaReachDense-v2 or v3
+        user_models = get_user_models(hf_username, unit["PandaReachDense-v3"], unit["library"])
+        if user_models == "":
+            print("Empty")
+            user_models = get_user_models(hf_username, unit["PandaReachDense-v2"], unit["library"])
     if unit["unit"] != "Unit 8 PII":
     # Get user model
         user_models = get_user_models(hf_username, unit['env'], unit['library'])