Skip to content

Commit aa8ef2e

Browse files
ndebuhrVaibhavs10
andauthored
Add CXR Foundation and Derm Foundation model libraries (Google Health AI) (#1049)
For two Google Health AI models, a primary usage pattern (with supporting documentation and notebooks) is downloading only the precomputed .npz embeddings. As such, I'm submitting for two new model libraries which are essentially an extension of the TF Keras model library, but they cleanly handle the non-`saved_model.pb` usage pattern in the `countDownload` override. This PR was designed to conform with https://huggingface.co/docs/hub/en/models-download-stats and https://huggingface.co/docs/hub/en/models-adding-libraries#register-your-library. Public models https://huggingface.co/google/cxr-foundation and https://huggingface.co/google/derm-foundation will be switched over once this PR is merged. --------- Co-authored-by: vb <[email protected]>
1 parent e660c35 commit aa8ef2e

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
150150
filter: false,
151151
countDownloads: `path:"adapter_config.json"`,
152152
},
153+
"cxr-foundation": {
154+
prettyLabel: "CXR Foundation",
155+
repoName: "cxr-foundation",
156+
repoUrl: "https://github.com/google-health/cxr-foundation",
157+
filter: false,
158+
countDownloads: `path:"precomputed_embeddings/embeddings.npz" OR path:"pax-elixr-b-text/saved_model.pb"`,
159+
},
153160
deepforest: {
154161
prettyLabel: "DeepForest",
155162
repoName: "deepforest",
@@ -173,6 +180,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
173180
snippets: snippets.depth_pro,
174181
filter: false,
175182
},
183+
"derm-foundation": {
184+
prettyLabel: "Derm Foundation",
185+
repoName: "derm-foundation",
186+
repoUrl: "https://github.com/google-health/derm-foundation",
187+
filter: false,
188+
countDownloads: `path:"scin_dataset_precomputed_embeddings.npz" OR path:"saved_model.pb"`,
189+
},
176190
diffree: {
177191
prettyLabel: "Diffree",
178192
repoName: "Diffree",

0 commit comments

Comments
 (0)