1aurent
/

phikon-distil-mobilenet_v2-kather2016

Image Classification

feature-extraction

Model card Files Files and versions

Metrics Training metrics Community

1aurent commited on Nov 9, 2023

Commit

437b7dc

·

1 Parent(s): b68b778

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -56,8 +56,9 @@ img = Image.open(
 )
 # load image_processor and model from the hub
-image_processor = AutoImageProcessor.from_pretrained("owkin/phikon")
-model = AutoModelForImageClassification.from_pretrained("1aurent/phikon-distil-mobilenet_v2-kather2016")
 inputs = image_processor(img, return_tensors="pt")
 outputs = model(**inputs)

 )
 # load image_processor and model from the hub
+model_name = "1aurent/phikon-distil-mobilenet_v2-kather2016"
+image_processor = AutoImageProcessor.from_pretrained(model_name)
+model = AutoModelForImageClassification.from_pretrained(model_name)
 inputs = image_processor(img, return_tensors="pt")
 outputs = model(**inputs)