londonaicentre · Seferin · Jun 2, 2026
diff --git a/Home.py b/Home.py
@@ -35,6 +35,8 @@
   schemas:
     - module: oncollamaschemav3
       root_class: OncoLlamaModel
+    - module: oncoschema.schema
+      root_class: OncologyModel
   ```
 
 **2. Prepare Prediction Files**

diff --git a/pages/__init__.py b/pages/__init__.py
diff --git a/requirements.txt b/requirements.txt
@@ -2,3 +2,4 @@ streamlit>=1.28.0
 pydantic>=2.0.0
 pyyaml>=6.0
 oncollamaschemav3 @ git+https://github.com/londonaicentre/oncollamaschemav3.git
+londonaicentre-oncoschema>=2.0.2
diff --git a/schemas.yaml b/schemas.yaml
@@ -8,3 +8,5 @@
 schemas:
   - module: oncollamaschemav3
     root_class: OncoLlamaModel
+  - module: oncoschema.schema
+    root_class: OncologyModel
diff --git a/utils/__init__.py b/utils/__init__.py