Update README.md

README.md

@@ -33,15 +33,18 @@ from urllib.request import urlopen
 from PIL import Image
 import timm
 
-img = Image.open(urlopen(
-  "https://github.com/owkin/HistoSSLscaling/raw/main/assets/example.tif"
-))
+# get example histology image
+img = Image.open(
+  urlopen(
+    "https://github.com/owkin/HistoSSLscaling/raw/main/assets/example.tif"
+  )
+)
 
+# load model from the hub
 model = timm.create_model(
-  "hf-hub:1aurent/vit_base_patch16_224.owkin_pancancer",
+  model_name="hf-hub:1aurent/vit_base_patch16_224.owkin_pancancer",
   pretrained=True,
-)
-model = model.eval()
+).eval()
 
 # get model specific transforms (normalization, resize)
 data_config = timm.data.resolve_model_data_config(model)