Set braindao model type automatically

app.py

@@ -171,7 +171,7 @@ with demo:
                     model_name_textbox = gr.Textbox(label="Model name")
                     revision_name_textbox = gr.Textbox(label="Revision commit", placeholder="main")
                     model_type = gr.Dropdown(
-                        choices=[t.to_str(" ") for t in ModelType if t not in [ModelType.Unknown, ModelType.BrainDAO]],
+                        choices=[t.to_str(" ") for t in ModelType if t in [ModelType.PT, ModelType.FT]],
                         label="Model type",
                         multiselect=False,
                         value=None,
@@ -195,8 +195,14 @@ with demo:
 
             submit_button = gr.Button("Submit")
             submission_result = gr.Markdown()
+            
+            def submit_with_braindao_check(model_name, revision, model_type):
+                if model_name.split("/")[0] == "braindao":
+                    model_type = ModelType.BrainDAO.to_str(" ")
+                return add_new_eval(model_name, revision, model_type)
+            
             submit_button.click(
-                add_new_eval,
+                submit_with_braindao_check,
                 [
                     model_name_textbox,
                     # base_model_name_textbox,