Update app.py

app.py

@@ -26,6 +26,7 @@ def add_new_eval(
     model_name_textbox: str,
     revision_name_textbox: str,
     model_link: str,
+    team_name: str,
     contact_email: str
 ):
     if input_file is None:
@@ -98,7 +99,10 @@ def add_new_eval(
             new_data.append(upload_data[key])
         else:
             new_data.append(0)
-    new_data.append("User Upload")
+    if team_name =='' or 'vbench' in team_name.lower():
+        new_data.append("User Upload")
+    else:
+        new_data.append(team_name)
     new_data.append(contact_email)
     csv_data.loc[col] = new_data
     csv_data = csv_data.to_csv(CSV_DIR, index=False)
@@ -541,6 +545,7 @@ with block:
                         model_name_textbox,
                         revision_name_textbox,
                         model_link,
+                        team_name,
                         contact_email
                     ],
                     outputs=[submit_button, submit_succ_button, fail_textbox]