tcr is back online

app.py

@@ -19,8 +19,7 @@ all_models = ModelRegistry()
 ppi_task = all_tasks.register_task(PpiTask(model_dict=all_models))
 tdi_task = all_tasks.register_task(DtiTask(model_dict=all_models))
 ps_task = all_tasks.register_task(PsTask(model_dict=all_models))
-# tcr_task = all_tasks.register_task(TcrTask(model_dict=all_models))
-tcr_task = "not ready yet"
+tcr_task = all_tasks.register_task(TcrTask(model_dict=all_models))
 
 # create the model holders. hold the model and the tokenizer, lazy download
 # note that the list of relevent tasks needs to be stated.
@@ -38,11 +37,18 @@ all_models.register_model(
 )
 all_models.register_model(
     model_path="ibm/biomed.omics.bl.sm.ma-ted-458m",
-    task_list=[ppi_task, tcr_task],
+    task_list=[ppi_task],
 )
-all_models.register_model("https://huggingface.co/ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_clintox_tox")
-all_models.register_model("https://huggingface.co/ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_clintox_fda")
-all_models.register_model("https://huggingface.co/ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_bbbp")
+all_models.register_model(
+    "https://huggingface.co/ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_clintox_tox"
+)
+all_models.register_model(
+    "https://huggingface.co/ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_clintox_fda"
+)
+all_models.register_model(
+    "https://huggingface.co/ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_bbbp"
+)
+
 
 def create_application():
     def task_change(value):
@@ -59,16 +65,16 @@ def create_application():
             )
         else:
             return (gr.update(visible=False), *visibility)
-    
+
     def model_change(value):
-        return gr.update(link=f"https://huggingface.co/{value}",visible=True)
+        return gr.update(link=f"https://huggingface.co/{value}", visible=True)
 
     with gr.Blocks(theme="Zarkel/IBM_Carbon_Theme") as application:
         task_dropdown = gr.Dropdown(
             choices=["Select task"] + list(all_tasks.keys()), label="Mammal Task"
         )
         task_dropdown.interactive = True
-        with gr.Row() :
+        with gr.Row():
             model_name_dropdown = gr.Dropdown(
                 choices=[
                     model_name
@@ -80,11 +86,11 @@ def create_application():
                 visible=False,
                 scale=10,
             )
-            goto_card_button = gr.Button("Link to model card",size='sm',scale=0,visible=False)
+            goto_card_button = gr.Button(
+                "Link to model card", size="sm", scale=0, visible=False
+            )
             model_name_dropdown.change(
-                model_change,
-                inputs=[model_name_dropdown],
-                outputs=[goto_card_button]
+                model_change, inputs=[model_name_dropdown], outputs=[goto_card_button]
             )
 
         task_dropdown.change(
@@ -96,7 +102,6 @@ def create_application():
                 for task in all_tasks
             ],
         )
-            
 
         # def set_demo_vis(main_text):
         #     main_text=main_text