Update yall.py

yall.py

@@ -96,19 +96,29 @@ def create_yall():
         else:
             model_name = model_id = 'Unknown'
 
+        # Calculate the average
+        try:
+            agieval = float(values[1].strip())
+            gpt4all = float(values[2].strip())
+            bigbench = float(values[4].strip())
+            average = (agieval + gpt4all + bigbench) / 3
+        except (ValueError, IndexError) as e:
+            print(f"Error processing values: {e}")
+            continue
+
         # Create a GistInfo object and add it to the list
         gist_info = GistInfo(
             gist_id=data_dict['id'],
             filename=filename,
-            url=data_dict['html_url'],  # Assuming html_url is the URL of the gist
+            url=data_dict['html_url'],
             model_name=model_name,
             model_id=model_id,
             model=values[0].strip(),
-            agieval=float(values[1].strip()),
-            gpt4all=float(values[2].strip()),
+            agieval=agieval,
+            gpt4all=gpt4all,
             truthfulqa=float(values[3].strip()),
-            bigbench=float(values[4].strip()),
-            average=float(sum([values[1],values[2],values[4]]/3.strip()),
+            bigbench=bigbench,
+            average=average,
         )
         gist_infos.append(gist_info)