moved creation and collection of tasks and models into memmal_demo

app.py

@@ -1,14 +1,6 @@
 import gradio as gr
 
-from mammal_demo.demo_framework import (
-    ModelRegistry,
-    TaskRegistry,
-)
-from mammal_demo.dti_task import DtiTask
-from mammal_demo.ppi_task import PpiTask
-from mammal_demo.ps_task import PsTask
-from mammal_demo.tcr_task import TcrTask
-
+import mammal_demo
 MAIN_MARKDOWN_TEXT = """
 
 The **[ibm/biomed.omics.bl.sm.ma-ted-458m](https://huggingface.co/models?sort=trending&search=ibm%2Fbiomed.omics.bl)**  model family is a biomedical foundation model and its finetuned variants trained on over 2 billion biological samples across multiple modalities, including proteins, small molecules, and single-cell gene data.
@@ -20,51 +12,7 @@ The syntax allows for dynamic combinations of tokens and scalars, enabling class
 This page demonstraits a variety of drug discovery and biomedical tasks for the model family.  Select the task to access the specific demos.
 """
 
-
-all_tasks = TaskRegistry()
-all_models = ModelRegistry()
-
-# first create the required tasks
-# Note that the tasks need access to the models, as the model to use depends on the state of the widget
-# we pass the all_models dict and update it when we actualy have the models.
-
-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))
-
-# create the model holders. hold the model and the tokenizer, lazy download
-# note that the list of relevent tasks needs to be stated.
-all_models.register_model(
-    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m.dti_bindingdb_pkd",
-    task_list=[tdi_task],
-)
-all_models.register_model(
-    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m.dti_bindingdb_pkd_peer",
-    task_list=[tdi_task],
-)
-
-all_models.register_model(
-    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m.tcr_epitope_bind",
-    task_list=[tcr_task],
-)
-all_models.register_model(
-    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m.protein_solubility",
-    task_list=[ps_task],
-)
-all_models.register_model(
-    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m",
-    task_list=[ppi_task],
-)
-all_models.register_model(
-    "ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_clintox_tox"
-)
-all_models.register_model(
-    "ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_clintox_fda"
-)
-all_models.register_model(
-    "ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_bbbp"
-)
+all_tasks, all_models = mammal_demo.tasks_and_models()
 
 
 def create_application():

mammal_demo/__init__.py

@@ -0,0 +1,58 @@
+
+from mammal_demo.demo_framework import (
+    ModelRegistry,
+    TaskRegistry,
+)
+from mammal_demo.dti_task import DtiTask
+from mammal_demo.ppi_task import PpiTask
+from mammal_demo.ps_task import PsTask
+from mammal_demo.tcr_task import TcrTask
+
+
+def tasks_and_models():
+    all_tasks = TaskRegistry()
+    all_models = ModelRegistry()
+
+# first create the required tasks
+# Note that the tasks need access to the models, as the model to use depends on the state of the widget
+# we pass the all_models dict and update it when we actualy have the models.
+
+    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))
+
+# create the model holders. hold the model and the tokenizer, lazy download
+# note that the list of relevent tasks needs to be stated.
+    all_models.register_model(
+    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m.dti_bindingdb_pkd",
+    task_list=[tdi_task],
+)
+    all_models.register_model(
+    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m.dti_bindingdb_pkd_peer",
+    task_list=[tdi_task],
+)
+
+    all_models.register_model(
+    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m.tcr_epitope_bind",
+    task_list=[tcr_task],
+)
+    all_models.register_model(
+    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m.protein_solubility",
+    task_list=[ps_task],
+)
+    all_models.register_model(
+    model_path="ibm/biomed.omics.bl.sm.ma-ted-458m",
+    task_list=[ppi_task],
+)
+    all_models.register_model(
+    "ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_clintox_tox"
+)
+    all_models.register_model(
+    "ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_clintox_fda"
+)
+    all_models.register_model(
+    "ibm/biomed.omics.bl.sm.ma-ted-458m.moleculenet_bbbp"
+)
+    
+    return all_tasks,all_models