⚡ adding new weights as defaults

rbturnbull · Jul 13, 2023 · e5e1543 · e5e1543
1 parent f7094a2
commit e5e1543
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "supercat"
-version = "0.2.1"
+version = "0.3.0"
 description = "A deep learning model for CT scan super-resolution."
 authors = ["MDAP <[email protected]>"]
 license = "Apache Software License 2.0"

diff --git a/supercat/apps.py b/supercat/apps.py
@@ -281,7 +281,9 @@ def pretrained_location(
         dim:int = ta.Param(default=2, help="The dimension of the dataset. 2 or 3."),
     ) -> str:
         assert dim in [2,3]
-        return f"https://github.com/rbturnbull/supercat/releases/download/v0.2.1/supercat-{dim}D.0.2.pkl"
+        if dim == 2:
+            return f"https://github.com/rbturnbull/supercat/releases/download/v0.2.1/supercat-{dim}D.0.2.pkl"
+        return f"https://github.com/rbturnbull/supercat/releases/download/v0.3.0/supercat-{dim}D.0.3.pkl"        
 
 
 class SupercatDiffusion(Supercat):
@@ -303,7 +305,9 @@ def pretrained_location(
         dim:int = ta.Param(default=2, help="The dimension of the dataset. 2 or 3."),
     ) -> str:
         assert dim in [2,3]
-        return f"https://github.com/rbturnbull/supercat/releases/download/v0.2.1/supercat-diffusion-{dim}D.0.2.pkl"
+        if dim == 2:
+            return f"https://github.com/rbturnbull/supercat/releases/download/v0.2.1/supercat-diffusion-{dim}D.0.2.pkl"
+        return f"https://github.com/rbturnbull/supercat/releases/download/v0.3.0/supercat-diffusion-{dim}D.0.3.pkl"
 
     # def output_results(
     #     self,