store model in releases

UC-Berkeley-I-School · Apr 19, 2024 · c2f041b · c2f041b
1 parent 7cce7c1
commit c2f041b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.github/workflows/deploy.yml b/.github/workflows/deploy.yml
@@ -73,6 +73,12 @@ jobs:
       - name: Install dependencies
         run: ${{ steps.detect-package-manager.outputs.manager }} ${{ steps.detect-package-manager.outputs.command }}
 
+      - uses: suisei-cn/[email protected]
+        name: Download the models
+        with:
+          url: https://github.com/UC-Berkeley-I-School/mids-210-medfusion-analytics-spring24/releases/download/v1.0.0/bioclinicalbert_nlp_cls_best_fp16.onnx
+          target: model/onnx/
+
       - name: Build with Next.js
         run: ${{ steps.detect-package-manager.outputs.runner }} next build
 

diff --git a/utils/constants.ts b/utils/constants.ts
@@ -2,7 +2,8 @@ const GITHUB_PAGE_URL = "https://github.com/UC-Berkeley-I-School/mids-210-medfus
 const RELEASE_VERSION = "v0.0.1-alpha/";
 
 // export const NLP_MODEL_URL = "https://mids-210-medical-multi-modal-models.s3.us-west-2.amazonaws.com/bioclinicalbert_nlp_cls_best.onnx";
-export const NLP_MODEL_URL = "https://mids-210-medical-multi-modal-models.s3.us-west-2.amazonaws.com/bioclinicalbert_nlp_cls_best_fp16.onnx";
+// export const NLP_MODEL_URL = "https://mids-210-medical-multi-modal-models.s3.us-west-2.amazonaws.com/bioclinicalbert_nlp_cls_best_fp16.onnx";
+export const NLP_MODEL_URL = "./bioclinicalbert_nlp_cls_best_fp16.onnx";
 export const IMAGE_MODEL_URL = "./EfficientNet-B3_Model.onnx";
 export const TABULAR_MODEL_URL = "./pipeline_tabular_model.onnx";
 export const FUSION_MODEL_URL = './fusion_model_1a.onnx';