Update target_dist.json loading from HuggingFace

Author: Theodore Zhao

inference_utils/output_processing.py

@@ -2,8 +2,7 @@
 from scipy import stats
 import numpy as np
 
-
-target_dist = json.load(open("inference_utils/target_dist.json"))
+import huggingface_hub
 
 
 def check_mask_stats(img, mask, modality_type, target):
@@ -12,6 +11,9 @@ def check_mask_stats(img, mask, modality_type, target):
     # modality_type: str, see target_dist.json for the list of modality types
     # target: str, see target_dist.json for the list of targets
 
+    huggingface_hub.hf_hub_download('microsoft/BiomedParse', filename='target_dist.json', local_dir='./inference_utils')
+    target_dist = json.load(open("inference_utils/target_dist.json"))
+    
     if modality_type not in target_dist:
         raise ValueError(f"Currently support modality types: {list(target_dist.keys())}")
 

inference_utils/target_dist.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:384e5f8db866b757dc61e28d205c1e7223bf0b7ed416f91289ad0821a78e52db
-size 10812
+oid sha256:1ef667c043867efc28b0c217448f99c0a8f24f11ffceb02923eaf7e4c39038ed
+size 11183