philbrik commited on
Commit
3f2d8dc
·
verified ·
1 Parent(s): a3e8b4b

Update pete_predictor_v2.py

Browse files

Load model from hugging face

Files changed (1) hide show
  1. pete_predictor_v2.py +2 -1
pete_predictor_v2.py CHANGED
@@ -17,6 +17,7 @@
17
  import functools
18
  from typing import Any, Mapping
19
 
 
20
  from ez_wsi_dicomweb import credential_factory
21
  from ez_wsi_dicomweb import dicom_slide
22
  from ez_wsi_dicomweb import patch_embedding
@@ -33,7 +34,7 @@ from data_models import embedding_converter
33
 
34
 
35
  def _load_huggingface_model() -> tf.keras.Model:
36
- raise ValueError('Liron implement model loader')
37
 
38
 
39
  def _endpoint_model(ml_model: tf.keras.Model, image: np.ndarray) -> np.ndarray:
 
17
  import functools
18
  from typing import Any, Mapping
19
 
20
+ from huggingface_hub import from_pretrained_keras
21
  from ez_wsi_dicomweb import credential_factory
22
  from ez_wsi_dicomweb import dicom_slide
23
  from ez_wsi_dicomweb import patch_embedding
 
34
 
35
 
36
  def _load_huggingface_model() -> tf.keras.Model:
37
+ return loaded_model = from_pretrained_keras("google/path-foundation")
38
 
39
 
40
  def _endpoint_model(ml_model: tf.keras.Model, image: np.ndarray) -> np.ndarray: